fixed issue for take_one_step