Update test case to use Driver Factory