1 files changed, 5 insertions, 2 deletions
diff --git a/tools/slang-test/test-context.h b/tools/slang-test/test-context.h
index cfa9837da..b07ca94e6 100644
--- a/tools/slang-test/test-context.h
+++ b/tools/slang-test/test-context.h
@@ -5,15 +5,18 @@
 
 #include "../../source/core/slang-string-util.h"
 #include "../../source/core/platform.h"
-#include "../../source/core/slang-app-context.h"
+#include "../../source/core/slang-std-writers.h"
 #include "../../source/core/dictionary.h"
+#include "../../source/core/slang-test-tool-util.h"
 
 #include "options.h"
 
 class TestContext
 {
     public:
-    typedef SlangResult(*InnerMainFunc)(Slang::AppContext* appContext, SlangSession* session, int argc, const char*const* argv);
+
+    typedef Slang::TestToolUtil::InnerMainFunc InnerMainFunc;
+
 
         /// Get the slang session
     SlangSession* getSession() const { return m_session;  }