Add error handling to driver loading