Attempt to fix test driver