diff --git a/test/cpp/src/TestClient.cpp b/test/cpp/src/TestClient.cpp
index 4764a4c..8ae819a 100644
--- a/test/cpp/src/TestClient.cpp
+++ b/test/cpp/src/TestClient.cpp
@@ -75,7 +75,7 @@
   }
 
 
-  shared_ptr<TTransport> transport;
+  shared_ptr<TBufferBase> transport;
 
   shared_ptr<TSocket> socket(new TSocket(host, port));
 
@@ -87,7 +87,8 @@
     transport = bufferedSocket;
   }
 
-  shared_ptr<TBinaryProtocol> protocol(new TBinaryProtocol(transport));
+  shared_ptr< TBinaryProtocolT<TBufferBase> > protocol(
+      new TBinaryProtocolT<TBufferBase>(transport));
   ThriftTestClient testClient(protocol);
 
   uint64_t time_min = 0;