diff --git a/plugins/ui/setup.py b/plugins/ui/setup.py
index d2d034634..fa2c2a39a 100644
--- a/plugins/ui/setup.py
+++ b/plugins/ui/setup.py
@@ -6,9 +6,10 @@
 js_dir = "src/js/"
 dest_dir = os.path.join("src/deephaven/ui/_js")
 
-# remove the build directory to ensure that the package is built from the latest js files
+# remove the build/dist directory to ensure that the package is built from the latest js files
 try:
     shutil.rmtree("build")
+    shutil.rmtree("dist")
 except FileNotFoundError:
     pass