Apply driver api change

Apply driver api changes introduced from https://github.com/NFLabs/zeppelin/pull/39