1 files changed, 4 insertions, 2 deletions
diff --git a/tools/slang-test/test-context.cpp b/tools/slang-test/test-context.cpp
index 8fa5cb345..2872ddb47 100644
--- a/tools/slang-test/test-context.cpp
+++ b/tools/slang-test/test-context.cpp
@@ -5,6 +5,8 @@
 #include "../../source/core/slang-string-util.h"
 #include "../../source/core/slang-shared-library.h"
 
+#include "../../source/core/slang-test-tool-util.h"
+
 #include <stdio.h>
 #include <stdlib.h>
 
@@ -15,14 +17,14 @@ TestContext::TestContext()
     m_session = nullptr;
 }
 
-Result TestContext::init()
+Result TestContext::init(const char* exePath)
 {
     m_session = spCreateSession(nullptr);
     if (!m_session)
     {
         return SLANG_FAIL;
     }
-
+    SLANG_RETURN_ON_FAIL(TestToolUtil::getExeDirectoryPath(exePath, exeDirectoryPath));
     return SLANG_OK;
 }