From 0696ea87b7ea4d3e4b520cbb58026f243b7b3d13 Mon Sep 17 00:00:00 2001
From: ricky <rickycancro@gmail.com>
Date: Mon, 11 Jul 2016 09:59:22 -0700
Subject: [PATCH] fixed trait example

---
 examples_extra/ASTraitCollection/Sample/KittenNode.m | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/examples_extra/ASTraitCollection/Sample/KittenNode.m b/examples_extra/ASTraitCollection/Sample/KittenNode.m
index 750bf63a1b..a7f95e91fb 100644
--- a/examples_extra/ASTraitCollection/Sample/KittenNode.m
+++ b/examples_extra/ASTraitCollection/Sample/KittenNode.m
@@ -155,13 +155,14 @@ static const CGFloat kInnerPadding = 10.0f;
 {
   OverrideViewController *overrideVC = [[OverrideViewController alloc] init];
   
+  __weak OverrideViewController *weakOverrideVC = overrideVC;
   overrideVC.overrideDisplayTraitsWithTraitCollection = ^(UITraitCollection *traitCollection) {
     ASTraitCollection *asyncTraitCollection = [ASTraitCollection traitCollectionWithDisplayScale:traitCollection.displayScale
                                                                               userInterfaceIdiom:traitCollection.userInterfaceIdiom
                                                                              horizontalSizeClass:UIUserInterfaceSizeClassCompact
                                                                                verticalSizeClass:UIUserInterfaceSizeClassCompact
                                                                             forceTouchCapability:traitCollection.forceTouchCapability
-                                                                          traitCollectionContext:nil];
+                                                                                   containerSize:weakOverrideVC.view.bounds.size];
     return asyncTraitCollection;
   };