fix game over logic