Fixed a mistake with my previous commit