diff --git a/uikit/CSSLayout/Tests/CSSLayoutTests.m b/uikit/CSSLayout/Tests/CSSLayoutTests.m
index e9ce9b1d..4eb54b19 100644
--- a/uikit/CSSLayout/Tests/CSSLayoutTests.m
+++ b/uikit/CSSLayout/Tests/CSSLayoutTests.m
@@ -21,7 +21,7 @@
   XCTAssertEqual(0, CSSNodeGetInstanceCount());
 
   UIView *view = [[UIView alloc] initWithFrame:CGRectZero];
-  [view css_setFlex:1];
+  [view css_setFlexBasis:1];
   XCTAssertEqual(1, CSSNodeGetInstanceCount());
   view = nil;
 
@@ -33,11 +33,11 @@
   XCTAssertEqual(0, CSSNodeGetInstanceCount());
 
   UIView *view = [[UIView alloc] initWithFrame:CGRectZero];
-  [view css_setFlex:1];
+  [view css_setFlexBasis:1];
 
   for (int i=0; i<10; i++) {
     UIView *subview = [[UIView alloc] initWithFrame:CGRectZero];
-    [subview css_setFlex:1];
+    [subview css_setFlexBasis:1];
     [view addSubview:subview];
   }
   XCTAssertEqual(11, CSSNodeGetInstanceCount());