1 files changed, 2 insertions, 4 deletions
diff --git a/uitest/src/com/vaadin/tests/components/AbstractComponentTest.java b/uitest/src/com/vaadin/tests/components/AbstractComponentTest.java
index c086e03ae0..8dae56b079 100644
--- a/uitest/src/com/vaadin/tests/components/AbstractComponentTest.java
+++ b/uitest/src/com/vaadin/tests/components/AbstractComponentTest.java
@@ -165,12 +165,10 @@ public abstract class AbstractComponentTest<T extends AbstractComponent>
         // This is only to be screenshot-compatible with Vaadin 6, where
         // invisible components cause spacing
         if (visible) {
-            log.setHeight(null);
-            log.setWidth(null);
+            log.removeStyleName("displaynone");
             log.setCaption((String) log.getData());
         } else {
-            log.setHeight("0px");
-            log.setWidth("0px");
+            log.addStyleName("displaynone");
             log.setCaption(null);
         }
     }