diff --git a/setup.py b/setup.py
index a04171a..46c6f56 100644
--- a/setup.py
+++ b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name="pytorchts",
-    version="0.3.0",
+    version="0.3.1",
     description="PyTorch Probabilistic Time Series Modeling framework",
     long_description=open("README.md").read(),
     long_description_content_type="text/markdown",
@@ -16,7 +16,7 @@
     python_requires=">=3.6",
     install_requires=[
         "torch>=1.7.0",
-        "gluonts",
+        "gluonts @ git+https://github.com/awslabs/gluon-ts.git@master#egg=gluonts",
         "holidays",
         "numpy~=1.16",
         "pandas~=1.1",
@@ -26,7 +26,6 @@
         "tensorboard",
         "wandb",
     ],
-    dependency_links=["git+https://github.com/awslabs/gluon-ts.git@master#egg=gluonts"],
     test_suite="tests",
     tests_require=["flake8", "pytest"],
 )