diff --git a/.github/test.py b/.github/test.py
index f3afe2e..b3caee8 100644
--- a/.github/test.py
+++ b/.github/test.py
@@ -11,18 +11,25 @@
 def test_index():
     response = client.get("/")
     assert response.status_code == 200
-    assert "Welcome" in response.text
+    assert "<title>Schizo Bots</title>" in response.text
+    assert '<button id="pause-button" class="pause-button">Pause</button>' in response.text
+    assert '<button id="new-chat-button" class="new-chat-button">New Chat</button>' in response.text
 
 def test_new_chat():
+    # Ensure the chat is not paused before starting a new chat
+    client.post("/pause")
+    
     response = client.post("/new-chat")
     assert response.status_code == 200
     assert response.json() == {"status": "new chat started"}
 
 def test_pause():
+    # Check initial pause state
     response = client.post("/pause")
     assert response.status_code == 200
     assert response.json()["paused"] is True
 
+    # Toggle pause state back to False
     response = client.post("/pause")
     assert response.status_code == 200
     assert response.json()["paused"] is False