clean up bug