Fix zake driver with latest release

Change-Id: I58c7dfa2d61821c335846684f21f82a5e6dfd8e8