Fixed header problem in test driver