A fluid rotor and a fluid stator. The fluid stator surrounds the fluid rotor. The fluid stator has an intake end and a discharge end. The fluid stator is shaped to be inserted into a wellbore. A shaft passes through a rotational axis of the fluid rotor. The shaft is attached to the fluid rotor to rotate in union with the fluid rotor. The shaft defines a central fluid passage that extends from the intake end of the fluid rotor to the discharge end of the fluid rotor. A balance piston surrounds the shaft. The balance piston extends from an outer surface of the shaft to an inner surface of the fluid stator. The balance piston is positioned at the intake end.