Allow :driver parameter to be in either String or Class