1 files changed, 4 insertions, 5 deletions
diff --git a/view/kernelcache/HeadlessPlugin.cpp b/view/kernelcache/HeadlessPlugin.cpp
index d14ccd6d..63f82d4d 100644
--- a/view/kernelcache/HeadlessPlugin.cpp
+++ b/view/kernelcache/HeadlessPlugin.cpp
@@ -1,11 +1,10 @@
 #include <binaryninjaapi.h>
-#include "KCView.h"
-#include "KernelCache.h"
+#include "KernelCacheView.h"
+#include "transformers/KernelCacheTransforms.h"
 
 #ifdef __cplusplus
 extern "C" {
 #endif
-	// extern void RegisterSharedCacheWorkflow();
 #ifdef __cplusplus
 }
 #endif
@@ -16,8 +15,8 @@ extern "C"
 
 	BINARYNINJAPLUGIN bool CorePluginInit()
 	{
-		InitKernelcache();
-		// RegisterSharedCacheWorkflow();
+		KernelCacheViewType::Register();
+		RegisterTransformers();
 		return true;
 	}
 }
 \ No newline at end of file