Driving tests set to return in third step