Fixed an error in my previous commit