diff --git a/.dockerignore b/.dockerignore
new file mode 100644
index 0000000..fc807ea
--- /dev/null
+++ b/.dockerignore
@@ -0,0 +1 @@
+src/riva/asrlib/decoder/tmp_graph_construction/
\ No newline at end of file
diff --git a/pyproject.toml b/pyproject.toml
index 2bb2d66..d5e911f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ requires = [
     "wheel",
     "cmake>=3.25",
     "ninja",
-    "nanobind",
+    "nanobind@git+https://github.com/galv/nanobind#egg=fix-batch-size-1-type-cast",
 ]
 
 build-backend = "setuptools.build_meta"
diff --git a/setup.py b/setup.py
index 94b5e3f..4058c8f 100644
--- a/setup.py
+++ b/setup.py
@@ -96,7 +96,7 @@ def build_extension(self, ext: setuptools.extension.Extension):
 setuptools.setup(
     python_requires='>=3.7',
     name='riva-asrlib-decoder',
-    version='0.3.0',
+    version='0.3.1',
     author='NVIDIA',
     author_email='dgalvez@nvidia.com',
     keywords='ASR, CUDA, WFST, Decoder',