Fixed stupid clean up bug