diff --git a/selenium/t/pages/main_toolbar.py b/selenium/t/pages/main_toolbar.py
index cdb49be2..495704b2 100644
--- a/selenium/t/pages/main_toolbar.py
+++ b/selenium/t/pages/main_toolbar.py
@@ -9,4 +9,4 @@ def rsconnect_publish(self):
 
     @property
     def rsconnect_notification(self):
-        return s(by.css("#notification_rsconnect"))
+        return s(by.css("#notification_rsconnect_jupyter"))
diff --git a/selenium/t/test_publish_source.py b/selenium/t/test_publish_source.py
index 781b52af..4ee57804 100644
--- a/selenium/t/test_publish_source.py
+++ b/selenium/t/test_publish_source.py
@@ -26,7 +26,7 @@ def setup(self, browser_config, jupyter_url, notebook, connect_url):
 
         # navigate to the notebook
         browser.open_url(jupyter_url + notebook)
-
+        MainToolBar().rsconnect_publish.should(be.visible)
         MainToolBar().rsconnect_publish.click()