Forgot a if statement.