missed some cleanup