small tweak

mobxjs · Apr 1, 2019 · fcd4161 · fcd4161
1 parent 17195c6
commit fcd4161
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/packages/mobx-state-tree/src/types/utility-types/reference.ts b/packages/mobx-state-tree/src/types/utility-types/reference.ts
@@ -552,14 +552,15 @@ export function safeReference<IT extends IAnyComplexType>(
     subType: IT,
     options?: ReferenceOptionsGetSet<IT>
 ): IMaybe<IReferenceType<IT>> & ISafeReference {
-    return maybe(
+    const ret = maybe(
         reference(subType, {
             ...options,
             onInvalidated(ev) {
                 ev.removeRef()
             }
         })
-    ) as any
+    )
+    return ret as typeof ret & ISafeReference
 }
 
 /** @hidden */