fix test driver