Fixing if statement.