diff --git a/core/tc_client_driver.c b/core/tc_client_driver.c
index 081d90ef9152382341d21c50e84801a960c24b7b..efe4906cd4b1254adc5c09564b654a1b065c4f1d 100644
--- a/core/tc_client_driver.c
+++ b/core/tc_client_driver.c
@@ -1199,7 +1199,11 @@ static int tc_ns_client_init(void)
 	ret = load_tz_shared_mem(g_dev_node);
 	if (ret != 0)
 		goto unmap_res_mem;
+#if (KERNEL_VERSION(6, 4, 0) <= LINUX_VERSION_CODE)
 	g_driver_class = class_create(TC_NS_CLIENT_DEV);
+#else
+	g_driver_class = class_create(THIS_MODULE, TC_NS_CLIENT_DEV);
+#endif
 	if (IS_ERR_OR_NULL(g_driver_class)) {
 		tloge("class create failed");
 		ret = -ENOMEM;