Fixed a mistake I made considering the changes