fix test driver take 4