Use context when needed
Added fake driver, restored base tests, fixed load driver issue