The present invention addresses method, apparatus and computer program product for controlling a Direct Learning Algorithm. Thereby, a power amplifier operating in a non-linear state is controlled. A signal to be amplified is input to a pre-distorter provided for compensating for non-linearity of the power amplifier. The pre-distorted output signal is forwarded from the pre-distorter to the power amplifier. Parameters of the pre-distorter are adapted in plural steps based on an error between a linearized signal output from the power amplifier and the signal to be amplified using an adaptive direct learning algorithm. It is detected whether the error diverges; and adapting of the parameters of the pre-distorter is stopped when it is determined that the error is diverging.