diff --git a/.github/workflows/workflow.yaml b/.github/workflows/workflow.yaml
index 90a9727..fbd010f 100644
--- a/.github/workflows/workflow.yaml
+++ b/.github/workflows/workflow.yaml
@@ -21,4 +21,4 @@ jobs:
     needs: act
     uses: arcalot/arcaflow-docsgen/.github/workflows/reusable_workflow.yaml@main
     with:
-      plugin_path: "arcaflow_plugin_wait/wait_plugin.py"
+      plugin_path: "arcaflow_plugin_wait/arcaflow_plugin_wait.py"
diff --git a/arcaflow_plugin_wait/wait_plugin.py b/arcaflow_plugin_wait/arcaflow_plugin_wait.py
similarity index 100%
rename from arcaflow_plugin_wait/wait_plugin.py
rename to arcaflow_plugin_wait/arcaflow_plugin_wait.py
diff --git a/pyproject.toml b/pyproject.toml
index c61690d..25068c6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -13,7 +13,7 @@ classifiers = [
 ]
 homepage = "https://github.com/arcalot/arcaflow-plugin-wait"
 packages = [
-   { include="wait_plugin.py", from="./arcaflow_plugin_wait"  },
+   { include="arcaflow_plugin_wait.py", from="./arcaflow_plugin_wait"  },
 ]
 
 [tool.poetry.dependencies]
diff --git a/tests/test_arcaflow_plugin_wait.py b/tests/test_arcaflow_plugin_wait.py
index 88cad31..ee8f3ea 100644
--- a/tests/test_arcaflow_plugin_wait.py
+++ b/tests/test_arcaflow_plugin_wait.py
@@ -4,7 +4,7 @@
 from arcaflow_plugin_sdk import plugin, predefined_schemas
 
 import time
-import wait_plugin
+import arcaflow_plugin_wait
 
 WAIT_TIME = 0.1
 SKIPPED_WAIT_TIME = 1.0
@@ -15,13 +15,13 @@ class WaitTest(unittest.TestCase):
     @staticmethod
     def test_serialization():
         plugin.test_object_serialization(
-            wait_plugin.InputParams(
+            arcaflow_plugin_wait.InputParams(
                 WAIT_TIME
             )
         )
 
         plugin.test_object_serialization(
-            wait_plugin.SuccessOutput(
+            arcaflow_plugin_wait.SuccessOutput(
                 "Waited {:0.2f} seconds after being scheduled to wait for"
                 " {} seconds.".format(WAIT_TIME, WAIT_TIME),
                 actual_wait_seconds=WAIT_TIME
@@ -29,7 +29,7 @@ def test_serialization():
         )
 
         plugin.test_object_serialization(
-            wait_plugin.ErrorOutput(
+            arcaflow_plugin_wait.ErrorOutput(
                 error="Aborted {:0.2f} seconds after being scheduled to wait"
                 " for {} seconds.".format(PREMATURE_TIME, WAIT_TIME),
                 actual_wait_seconds=PREMATURE_TIME
@@ -38,10 +38,10 @@ def test_serialization():
 
     def test_functional(self):
         # Test simple wait
-        input_params = wait_plugin.InputParams(
+        input_params = arcaflow_plugin_wait.InputParams(
             seconds=WAIT_TIME
         )
-        wait_step = wait_plugin.WaitStep()
+        wait_step = arcaflow_plugin_wait.WaitStep()
         output_id, output_data = wait_step.wait(input_params)
 
         self.assertEqual("success", output_id)
@@ -51,10 +51,10 @@ def test_functional(self):
             " seconds.".format(output_data.actual_wait_seconds, WAIT_TIME)
         )
         # Test cancellation
-        input_params = wait_plugin.InputParams(
+        input_params = arcaflow_plugin_wait.InputParams(
             seconds=SKIPPED_WAIT_TIME
         )
-        wait_step = wait_plugin.WaitStep()
+        wait_step = arcaflow_plugin_wait.WaitStep()
         wait_step.cancel_step(wait_step, predefined_schemas.cancelInput())
         start_time = time.time()
         output_id, output_data = wait_step.wait(input_params)