Fixing driver issue