diff --git a/test/test.js b/test/test.js
index 48f47fa..0bba9bb 100644
--- a/test/test.js
+++ b/test/test.js
@@ -116,7 +116,7 @@ describe('Application life cycle test', function () {
         await waitForElement(By.xpath('//input[@name="url_rss"]'));
         await browser.findElement(By.xpath('//input[@name="url_rss"]')).sendKeys(url);
         await browser.findElement(By.xpath('//form[@id="add_rss"]//button[text()="Add"]')).click();
-        await waitForElement(By.xpath('//div[@id="notification" and @class="notification good"]'));
+        await waitForElement(By.xpath('//div[@id="notification" and contains(@class, "good")]'));
     }
 
     async function addUser(username, password) {
@@ -125,7 +125,7 @@ describe('Application life cycle test', function () {
         await browser.findElement(By.id('new_user_name')).sendKeys(username);
         await browser.findElement(By.id('new_user_passwordPlain')).sendKeys(password);
         await browser.findElement(By.xpath('//button[text()="Create"]')).click();
-        await waitForElement(By.xpath('//div[@id="notification" and @class="notification good"]'));
+        await waitForElement(By.xpath('//div[@id="notification" and contains(@class, "good")]'));
     }
 
     async function enableApi() {