Fixed bug in end button. Fucking lint.