Fix driver spec test