Added another if statement.