Fix gen_step_fun() tests