diff --git a/Source/HarmonyLinkUE/Private/DataRefreshActor.cpp b/Source/HarmonyLinkUE/Private/DataRefreshActor.cpp
index 74b8734..fa7c26c 100644
--- a/Source/HarmonyLinkUE/Private/DataRefreshActor.cpp
+++ b/Source/HarmonyLinkUE/Private/DataRefreshActor.cpp
@@ -1,7 +1,7 @@
 // DataRefreshActor.cpp
 
 #include "DataRefreshActor.h"
-#include "HarmonyLinkUEBPLibrary.h"
+#include "HarmonyLinkClient.h"
 
 // Sets default values
 ADataRefreshActor::ADataRefreshActor()
@@ -50,8 +50,8 @@ void ADataRefreshActor::OnServerStatusChecked(bool connected)
 	if (connected)
 	{
 		// Refresh the data and update the cached information
-		UHarmonyLinkUEBPLibrary::RefreshAllInfo(FDeviceInfoCallback());
-		UHarmonyLinkUEBPLibrary::RefreshVersionInfo(FVersionInfoCallback());
+		UHarmonyLinkClient::RefreshAllInfo(FDeviceInfoCallback());
+		UHarmonyLinkClient::RefreshVersionInfo(FVersionInfoCallback());
 	}
 }
 
@@ -59,10 +59,10 @@ void ADataRefreshActor::OnServerStatusChecked(bool connected)
 void ADataRefreshActor::RefreshData()
 {
 	// Only refresh data if the Harmony Link library is available
-	if (UHarmonyLinkUEBPLibrary::StaticClass())
+	if (UHarmonyLinkClient::StaticClass())
 	{
 		// Check server status asynchronously
-		UHarmonyLinkUEBPLibrary::CheckServerStatus(ServerStatusCallback);
+		UHarmonyLinkClient::CheckServerStatus(ServerStatusCallback);
 	}
 }
 
diff --git a/Source/HarmonyLinkUE/Public/DataRefreshActor.h b/Source/HarmonyLinkUE/Public/DataRefreshActor.h
index a24c607..dc51438 100644
--- a/Source/HarmonyLinkUE/Public/DataRefreshActor.h
+++ b/Source/HarmonyLinkUE/Public/DataRefreshActor.h
@@ -4,7 +4,7 @@
 
 #include "CoreMinimal.h"
 #include "GameFramework/Actor.h"
-#include "HarmonyLinkUEBPLibrary.h"
+#include "HarmonyLinkClient.h"
 #include "DataRefreshActor.generated.h"
 
 UCLASS()