A method of controlling water heater pilot flame ignition includes receiving, by a controller of a water heater, a user input and controlling a pilot gas valve to start a gas flow to a pilot burner in response to the user input. The method further includes, in response to the user input, controlling, by the controller, an igniter to generate an ignition spark for lighting a pilot flame. The method also includes controlling, by the controller, the pilot gas valve to maintain the gas flow to the pilot burner if the pilot flame is lit.