Fixed a silly mistake with the previous commit