diff --git a/build.py b/build.py
index fdb7b47554..14301f843d 100755
--- a/build.py
+++ b/build.py
@@ -1849,11 +1849,11 @@ def core_build(
             os.path.join(repo_install_dir, "lib", "libtritonserver.so"),
             os.path.join(install_dir, "lib"),
         )
-    # [FIXME] Placing the Triton server wheel file in 'python' for now, should
-    # have been upload to pip registry and be able to install directly
+    # [FIXME] Placing the tritonserver and tritonfrontend wheel files in 'python' for now,
+    # should be uploaded to pip registry to be able to install directly
     cmake_script.mkdir(os.path.join(install_dir, "python"))
     cmake_script.cp(
-        os.path.join(repo_install_dir, "python", "tritonserver*.whl"),
+        os.path.join(repo_install_dir, "python", "triton*.whl"),
         os.path.join(install_dir, "python"),
     )