fix test driver take 3