fixed the end button