diff --git a/AXWebViewController/AXWebViewController/AXWebViewController.m b/AXWebViewController/AXWebViewController/AXWebViewController.m
index a526928..94e6c0f 100644
--- a/AXWebViewController/AXWebViewController/AXWebViewController.m
+++ b/AXWebViewController/AXWebViewController/AXWebViewController.m
@@ -1480,7 +1480,8 @@ - (AXWebViewController *)ax_webViewController {
 }
 
 - (void)setAx_webViewController:(AXWebViewController *)ax_webViewController {
-    objc_setAssociatedObject(self, @selector(ax_webViewController), ax_webViewController, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
+    // Using assign to fix issue: https://github.com/devedbox/AXWebViewController/issues/23
+    objc_setAssociatedObject(self, @selector(ax_webViewController), ax_webViewController, OBJC_ASSOCIATION_ASSIGN);
 }
 @end
 #endif