diff --git a/onnx/convert_onnx.py b/onnx/convert_onnx.py
index bbb400d..7b4a7b9 100644
--- a/onnx/convert_onnx.py
+++ b/onnx/convert_onnx.py
@@ -1,5 +1,6 @@
 import math
 import argparse
+import sys
 
 import torch
 from torch import nn
@@ -11,6 +12,7 @@
 from detectron2.checkpoint import DetectionCheckpointer
 from detectron2.config import get_cfg
 
+sys.path.append(".")
 from sparseinst import add_sparse_inst_config