diff --git a/tools/profile_nvfp4_pack.py b/tools/profile_nvfp4_pack.py
index 02b2b25..4fc0636 100644
--- a/tools/profile_nvfp4_pack.py
+++ b/tools/profile_nvfp4_pack.py
@@ -5,10 +5,13 @@ from __future__ import annotations
 import argparse
 import json
 import os
+import sys
 import time
 import warnings
 from pathlib import Path
 
+sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
+
 warnings.filterwarnings("ignore", message="Found GPU0 NVIDIA GB10 which is of cuda capability 12.1.*", category=UserWarning)
 
 import torch