minor bugfix in test driver