fix: mocking.

This patch fixes mocking that was broken by a previous commit to make
the real datastax driver work