Fixed dumb bug added by last commit