diff --git a/Dockerfile b/Dockerfile
index 123081c85..c9c99b31a 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -10,7 +10,7 @@ COPY --chown=postgres:postgres . /app
 WORKDIR /app/pgml
 
 # Install pgml extension globally.
-RUN python3 setup.py install
+RUN pip3 install pgml
 
 # Listen on 0.0.0.0 and allow 'root' to connect without a password.
 # Please modify for production deployments accordingly.
diff --git a/pgml/pgml/__init__.py b/pgml/pgml/__init__.py
index 220b8561d..1342ac7bd 100644
--- a/pgml/pgml/__init__.py
+++ b/pgml/pgml/__init__.py
@@ -1,2 +1,2 @@
 def version():
-    return "0.1"
+    return "0.3"
diff --git a/pgml/setup.py b/pgml/setup.py
index 7c9ef7ade..d40f6ba96 100644
--- a/pgml/setup.py
+++ b/pgml/setup.py
@@ -1,17 +1,18 @@
 import setuptools
+from pgml import version
 
 with open("README.md", "r") as fh:
     long_description = fh.read()
 
 setuptools.setup(
     name="pgml",
-    version="0.1",
+    version=version(),
     author="PostgresML",
     author_email="hello@postgresml.com",
     description="Run machine learning inside Postgres.",
     long_description=long_description,
     long_description_content_type="text/markdown",
-    url="https://github.com/levkk/postgresml",
+    url="https://github.com/postgresml/postgresml",
     install_requires=[
         "sklearn",
     ],