Slight improvement of code to be tested.