fixed issues with my last commit