Fixed up an if statement that I messed up on