fixing 1 more issue with my clean up