The present invention relates to an arrangement for cooling of an electrical machine. The electrical machine comprises a rotor rotatably arranged around a rotation axis, a stator including a stator winding arranged radially outside of the rotor, a housing enclosing the rotor and the stator, at least one drain hole configured to drain a cooling fluid from the housing, at least one spraying device configured to spray cooling fluid on the stator winding and a pump configured to pump cooling fluid to the spraying device. The arrangement comprises a control unit, which is configured, when the electrical machine is in operation, to receive information of the temperature in at least one position of the stator winding and to control the pump such that it pumps a flow rate of the cooling fluid to the spraying device, which is related to the estimated temperature of the stator winding.