diff --git a/pip_package/build_pip_package.sh b/pip_package/build_pip_package.sh
index 958e1c73..3facb9f7 100755
--- a/pip_package/build_pip_package.sh
+++ b/pip_package/build_pip_package.sh
@@ -28,7 +28,7 @@ mkdir -p "$dest"
 
 source_dir="$PWD/.."
 
-pushd ../lit_nlp/client
+pushd ../lit_nlp
 yarn && yarn build
 popd
 
diff --git a/pip_package/setup.py b/pip_package/setup.py
index db84395d..eac8c911 100644
--- a/pip_package/setup.py
+++ b/pip_package/setup.py
@@ -38,7 +38,7 @@
 
 setuptools.setup(
     name="lit-nlp",
-    version="0.1.1",
+    version="0.2",
     description="Language Interpretability Tool.",
     long_description=long_description,
     long_description_content_type="text/markdown",
@@ -47,5 +47,10 @@
     packages=setuptools.find_packages(),
     license="Apache 2.0",
     install_requires=REQUIRED_PACKAGES,
-    package_data={"lit_nlp": ["client/build/*", "client/build/static/*"]},
+    package_data={
+        "lit_nlp": [
+            "client/build/*", "client/build/default/*",
+            "client/build/default/static/*"
+        ]
+    },
 )