Minor bug on survival mode