diff --git a/docker-compose.yml b/docker-compose.yml
index 42f2c6a..5960c99 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -40,7 +40,7 @@ services:
       dockerfile: Dockerfile.publishing_tools
       args:
         REPO_URL: https://github.com/SNEWS2/SNEWS_Publishing_Tools.git
-        BRANCH: test-framework #main
+        BRANCH: main
         HOP_USERNAME: ${HOP_USERNAME}
         HOP_PASSWORD: ${HOP_PASSWORD}
     container_name: publishing_tools_publisher
diff --git a/pyproject.toml b/pyproject.toml
index bc89a6b..21bcbc6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -22,7 +22,7 @@ psycopg2-binary = "^2.9"  # PostgreSQL adapter
 confluent-kafka = "^2.3" # Kafka client
 
 [tool.poetry.group.dev.dependencies]
-snews-pt = "^1.0.0"
+snews-pt = "^2.0.0"
 autopep8 = "^2.0.4"
 flake8 = "^6.1.0"
 pytest = "<6.3.0"  # Constrained by snews_pt