diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py
index c0bb9230f..a14ad4316 100644
--- a/test/smoke_test/smoke_test.py
+++ b/test/smoke_test/smoke_test.py
@@ -9,6 +9,7 @@
 import torch._dynamo
 import torch.nn as nn
 import torch.nn.functional as F
+from pathlib import Path
 
 gpu_arch_ver = os.getenv("MATRIX_GPU_ARCH_VERSION")
 gpu_arch_type = os.getenv("MATRIX_GPU_ARCH_TYPE")
@@ -16,7 +17,7 @@
 stable_version = os.getenv("MATRIX_STABLE_VERSION")
 package_type = os.getenv("MATRIX_PACKAGE_TYPE")
 target_os = os.getenv("TARGET_OS")
-BASE_DIR =  Path(__file__).parent.parent
+BASE_DIR =  Path(__file__).parent.parent.parent
 
 is_cuda_system = gpu_arch_type == "cuda"
 NIGHTLY_ALLOWED_DELTA = 3
@@ -84,6 +85,8 @@ def check_version(package: str) -> None:
                     raise RuntimeError(
                         f"{module['name']} version mismatch, expected {release_version[module['name']]} for channel {channel}. But its {module_version}"
                     )
+                else:
+                     print(f"{module['name']} version actual: {module_version} expected: {release_version[module['name']]} for channel {channel}.")
 
     else:
         print(f"Skip version check for channel {channel} as stable version is None")