diff --git a/pyproject.toml b/pyproject.toml
index 0eb9cc8..24f1b2b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -61,10 +61,9 @@ dependencies = [
 ]
 
 [project.urls]
-"Homepage" = "https://github.com/ultralytics/thop"
-
-[tool.setuptools]
-package-data = { "thop" = ["__version__.py"] }
+"Bug Reports" = "https://github.com/ultralytics/thop/issues"
+"Funding" = "https://ultralytics.com"
+"Source" = "https://github.com/ultralytics/thop/"
 
 [tool.setuptools.dynamic]
-version = { attr = "thop.__version__.VERSION" }
+version = { attr = "thop.__version__" }
diff --git a/thop/__init__.py b/thop/__init__.py
index 5f8f854..c1002e4 100644
--- a/thop/__init__.py
+++ b/thop/__init__.py
@@ -5,4 +5,5 @@
 from .utils import clever_format
 
 default_dtype = torch.float64
-__version__ = "0.2.0"
+
+__version__ = "0.2.1"