Fixed bug where second play through would end prematurely