Fixed an issue with my last commit