Another fix for if statement.