diff --git a/src/test/java/com/wikia/webdriver/elements/mercury/old/CommentsPageObject.java b/src/test/java/com/wikia/webdriver/elements/mercury/old/CommentsPageObject.java
index cc6bfbe024..8c3b0091dc 100644
--- a/src/test/java/com/wikia/webdriver/elements/mercury/old/CommentsPageObject.java
+++ b/src/test/java/com/wikia/webdriver/elements/mercury/old/CommentsPageObject.java
@@ -180,7 +180,7 @@ public boolean isMediaThumbnailInComment(String mediaType, int index) {
     WebElement mediaInComment;
     allComments.stream().forEach(e -> wait.forElementClickable(e));
     if ("Video".equals(mediaType)) {
-      mediaInComment = allComments.get(index).findElement(By.cssSelector("figure.comment-video"));
+      mediaInComment = allComments.get(index).findElement(By.cssSelector("figure.thumb"));
     } else {
       mediaInComment = allComments.get(index).findElement(By.cssSelector("figure"));
     }
@@ -191,7 +191,7 @@ public boolean isMediaLinkInComment(String mediaType, int index) throws WebDrive
     WebElement mediaInComment;
     allComments.stream().forEach(e -> wait.forElementClickable(e));
     if ("Video".equals(mediaType)) {
-      mediaInComment = allComments.get(index).findElement(By.cssSelector("figure.comment-video"));
+      mediaInComment = allComments.get(index).findElement(By.cssSelector("figure.thumb"));
     } else {
       mediaInComment = allComments.get(index).findElement(By.cssSelector("figure"));
     }