1 files changed, 11 insertions, 12 deletions

diff --git a/test/gtest/test_tle_udp_dev.h b/test/gtest/test_tle_udp_dev.h
index 02f544d..3eeef3a 100644
--- a/test/gtest/test_tle_udp_dev.h
+++ b/test/gtest/test_tle_udp_dev.h
@@ -21,8 +21,7 @@
 #include <gtest/gtest.h>
 
 #include <rte_errno.h>
-
-#include <tle_udp_impl.h>
+#include <tle_ctx.h>
 
 #define RX_NO_OFFLOAD 0
 #define TX_NO_OFFLOAD 0
@@ -32,11 +31,11 @@ using namespace std;
 class udp_dev : public ::testing::Test {
 
 public:
-	struct tle_udp_ctx *ctx;
-	struct tle_udp_dev *dev;
-	struct tle_udp_ctx_param prm;
-	struct tle_udp_dev_param dev_prm;
-	vector<tle_udp_dev*> devs;
+	struct tle_ctx *ctx;
+	struct tle_dev *dev;
+	struct tle_ctx_param prm;
+	struct tle_dev_param dev_prm;
+	vector<tle_dev *> devs;
 
 	virtual void SetUp(void)
 	{
@@ -56,16 +55,16 @@ public:
 		inet_pton(AF_INET6, "fe80::21e:67ff:fec2:2568",
 				&(dev_prm).local_addr6);
 
-		ctx = tle_udp_create(&prm);
+		ctx = tle_ctx_create(&prm);
 		ASSERT_NE(ctx, (void *) NULL);
 	}
 
 	virtual void TearDown(void)
 	{
-		for(auto d : devs) {
-			tle_udp_del_dev(d);
-		}
-		tle_udp_destroy(ctx);
+		for (auto d : devs)
+			tle_del_dev(d);
+
+		tle_ctx_destroy(ctx);
 	}
 };