Forgot to fix an if.