diff --git a/test/simulator/hei_sim_main.cpp b/test/simulator/hei_sim_main.cpp
index e83a577..b3291aa 100644
--- a/test/simulator/hei_sim_main.cpp
+++ b/test/simulator/hei_sim_main.cpp
@@ -1,19 +1,19 @@
-#include "hei_isolator.hpp"
+#include <hei_main.hpp>
 
 int main()
 {
-    libhei::Isolator iso;
-
     void * buffer = nullptr;
     size_t sz_buffer = 0;
 
-    iso.initialize( buffer, sz_buffer );
+    libhei::initialize( buffer, sz_buffer );
 
     void * chip = nullptr;
     libhei::ChipType_t chipType = 0;
-    libhei::IsolationData isoData;
+    libhei::IsolationData isoData { chip, chipType };
 
-    iso.isolate( chip, chipType, isoData );
+    libhei::isolate( isoData );
+
+    libhei::uninitialize();
 
     return 0;
 }