update to match new driver method