updated test to reflect driver changes