diff --git a/doc/html/hummingbird/index.html b/doc/html/hummingbird/index.html
index 37934e527..e045bde8e 100644
--- a/doc/html/hummingbird/index.html
+++ b/doc/html/hummingbird/index.html
@@ -38,7 +38,7 @@ <h1 class="title">Package <code>hummingbird</code></h1>
 Entrypoint for Hummingbird modules.
 &#34;&#34;&#34;
 
-__version__ = &#34;0.0.2&#34;
+__version__ = &#34;0.0.3&#34;
 __author__ = &#34;Microsoft&#34;
 __producer__ = &#34;hummingbird&#34;
 __producer_version__ = __version__
diff --git a/doc/html/hummingbird/ml/ir_converters/linked_node.html b/doc/html/hummingbird/ml/ir_converters/linked_node.html
index 788075cfa..2c7e50af5 100644
--- a/doc/html/hummingbird/ml/ir_converters/linked_node.html
+++ b/doc/html/hummingbird/ml/ir_converters/linked_node.html
@@ -114,12 +114,13 @@ <h1 class="title">Module <code>hummingbird.ml.ir_converters.linked_node</code></
                 # Convert the model from PyTorch to ONNX.
                 if constants.N_FEATURES not in extra_config:
                     extra_config[constants.ONNX_INPUTS] = all_tensors
-                # Early version of pytorch have a bug with exporting gemm into ONNX.
-                trimmed_version = torch.__version__.split(&#34;+&#34;)[0]  # For Win pytorch has a +cpu or +gpu postfix
-                torch_version = LooseVersion(trimmed_version)
-                gemm_min = LooseVersion(&#34;1.5.0&#34;)
-                if torch_version &lt;= gemm_min:
-                    extra_config[constants.TREE_IMPLEMENTATION] = &#34;tree_trav&#34;
+                # Pytorch has a bug with exporting gemm into ONNX.
+                # For the moment only tree_trav is enabled
+                # trimmed_version = torch.__version__.split(&#34;+&#34;)[0]  # For Win pytorch has a +cpu or +gpu postfix
+                # torch_version = LooseVersion(trimmed_version)
+                # gemm_min = LooseVersion(&#34;1.5.1&#34;)
+                # if torch_version &lt;= gemm_min:
+                extra_config[constants.TREE_IMPLEMENTATION] = &#34;tree_trav&#34;
                 pytorch_model = converter(node_, extra_config=extra_config)
 
                 # Generate the inputs for the tracing.
@@ -444,12 +445,13 @@ <h2 id="returns">Returns</h2>
                 # Convert the model from PyTorch to ONNX.
                 if constants.N_FEATURES not in extra_config:
                     extra_config[constants.ONNX_INPUTS] = all_tensors
-                # Early version of pytorch have a bug with exporting gemm into ONNX.
-                trimmed_version = torch.__version__.split(&#34;+&#34;)[0]  # For Win pytorch has a +cpu or +gpu postfix
-                torch_version = LooseVersion(trimmed_version)
-                gemm_min = LooseVersion(&#34;1.5.0&#34;)
-                if torch_version &lt;= gemm_min:
-                    extra_config[constants.TREE_IMPLEMENTATION] = &#34;tree_trav&#34;
+                # Pytorch has a bug with exporting gemm into ONNX.
+                # For the moment only tree_trav is enabled
+                # trimmed_version = torch.__version__.split(&#34;+&#34;)[0]  # For Win pytorch has a +cpu or +gpu postfix
+                # torch_version = LooseVersion(trimmed_version)
+                # gemm_min = LooseVersion(&#34;1.5.1&#34;)
+                # if torch_version &lt;= gemm_min:
+                extra_config[constants.TREE_IMPLEMENTATION] = &#34;tree_trav&#34;
                 pytorch_model = converter(node_, extra_config=extra_config)
 
                 # Generate the inputs for the tracing.
diff --git a/hummingbird/__init__.py b/hummingbird/__init__.py
index c709de4b6..66136b03a 100644
--- a/hummingbird/__init__.py
+++ b/hummingbird/__init__.py
@@ -8,7 +8,7 @@
 Entrypoint for Hummingbird modules.
 """
 
-__version__ = "0.0.2"
+__version__ = "0.0.3"
 __author__ = "Microsoft"
 __producer__ = "hummingbird"
 __producer_version__ = __version__
diff --git a/setup.cfg b/setup.cfg
index 907fe2bd7..fbddcd312 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 0.0.2
+current_version = 0.0.3
 
 [bdist_wheel]
 universal = 1
diff --git a/setup.py b/setup.py
index 7523e590e..26c718a64 100644
--- a/setup.py
+++ b/setup.py
@@ -9,7 +9,6 @@
 assert packages
 
 # read version from the package file.
-version_str = "0.0.2"
 with (open(os.path.join(this, "hummingbird/__init__.py"), "r")) as f:
     line = [_ for _ in [_.strip("\r\n ") for _ in f.readlines()] if _.startswith("__version__")]
     if len(line) > 0: