fixed bug in the driver class