diff --git a/setup.py b/setup.py
index d9231899..66326d46 100644
--- a/setup.py
+++ b/setup.py
@@ -31,6 +31,7 @@
     url="https://github.com/KevinMusgrave/pytorch-metric-learning",
     package_dir={"": "src"},
     packages=setuptools.find_packages(where="src"),
+    license_files=('LICENSE',),
     classifiers=[
         "Programming Language :: Python :: 3",
         "License :: OSI Approved :: MIT License",
diff --git a/src/pytorch_metric_learning/__init__.py b/src/pytorch_metric_learning/__init__.py
index 3d67cd6b..54499df3 100644
--- a/src/pytorch_metric_learning/__init__.py
+++ b/src/pytorch_metric_learning/__init__.py
@@ -1 +1 @@
-__version__ = "2.4.0"
+__version__ = "2.4.1"