diff --git a/reagent-data-views/src/reagent_data_views/server/core.clj b/reagent-data-views/src/reagent_data_views/server/core.clj
index cf03c27..ccf9406 100644
--- a/reagent-data-views/src/reagent_data_views/server/core.clj
+++ b/reagent-data-views/src/reagent_data_views/server/core.clj
@@ -43,4 +43,5 @@
 
 (defn set-context-fn!
   [^Atom view-system f]
-  (swap! view-system assoc-in [:reagent-data-views :context-fn] f))
\ No newline at end of file
+  (swap! view-system assoc-in [:reagent-data-views :context-fn] f)
+  view-system)
\ No newline at end of file