Fix bucket logic when playing in survival