Small patch for Endless Mode