2 files changed, 4 insertions, 2 deletions
diff --git a/src/test/kotlin/test/pl/treksoft/kvision/core/WidgetWrapperSpec.kt b/src/test/kotlin/test/pl/treksoft/kvision/core/WidgetWrapperSpec.kt
index 41c5e346..a053627d 100644
--- a/src/test/kotlin/test/pl/treksoft/kvision/core/WidgetWrapperSpec.kt
+++ b/src/test/kotlin/test/pl/treksoft/kvision/core/WidgetWrapperSpec.kt
@@ -1,7 +1,7 @@
 package test.pl.treksoft.kvision.core
 
 import pl.treksoft.kvision.core.Root
-import pl.treksoft.kvision.css.UNIT
+import pl.treksoft.kvision.core.UNIT
 import pl.treksoft.kvision.core.Widget
 import pl.treksoft.kvision.core.WidgetWrapper
 import test.pl.treksoft.kvision.DomSpec
diff --git a/src/test/kotlin/test/pl/treksoft/kvision/html/ButtonSpec.kt b/src/test/kotlin/test/pl/treksoft/kvision/html/ButtonSpec.kt
index 5d632b59..681b3db3 100644
--- a/src/test/kotlin/test/pl/treksoft/kvision/html/ButtonSpec.kt
+++ b/src/test/kotlin/test/pl/treksoft/kvision/html/ButtonSpec.kt
@@ -15,7 +15,9 @@ class ButtonSpec : DomSpec {
     fun render() {
         run {
             val root = Root("test")
-            val button = Button("Cancel", "fa-bars", BUTTONSTYLE.PRIMARY, BUTTONSIZE.LARGE, true)
+            val button = Button("Cancel", "fa-bars", BUTTONSTYLE.PRIMARY)
+            button.size = BUTTONSIZE.LARGE
+            button.block = true
             root.add(button)
             val element = document.getElementById("test")
             assertEquals("<button class=\"btn btn-primary btn-lg btn-block\" type=\"button\"><i class=\"fa fa-bars fa-lg\"></i> Cancel</button>", element?.innerHTML, "Should render correct html button")