diff --git a/libraries/botbuilder-dialogs/setup.py b/libraries/botbuilder-dialogs/setup.py
index 0525ca6f5..236618573 100644
--- a/libraries/botbuilder-dialogs/setup.py
+++ b/libraries/botbuilder-dialogs/setup.py
@@ -44,6 +44,9 @@
         "botbuilder.dialogs.prompts",
         "botbuilder.dialogs.choices",
         "botbuilder.dialogs.skills",
+        "botbuilder.dialogs.memory",
+        "botbuilder.dialogs.memory.path_resolvers",
+        "botbuilder.dialogs.memory.scopes",
     ],
     install_requires=REQUIRES + TEST_REQUIRES,
     tests_require=TEST_REQUIRES,
diff --git a/libraries/botbuilder-schema/botbuilder/schema/teams/__init__.py b/libraries/botbuilder-schema/botbuilder/schema/teams/__init__.py
index a4feffa36..86872f87d 100644
--- a/libraries/botbuilder-schema/botbuilder/schema/teams/__init__.py
+++ b/libraries/botbuilder-schema/botbuilder/schema/teams/__init__.py
@@ -63,6 +63,7 @@
 from ._models_py3 import MeetingParticipantInfo
 from ._models_py3 import CacheInfo
 from ._models_py3 import TabContext
+from ._models_py3 import TabEntityContext
 from ._models_py3 import TabRequest
 from ._models_py3 import TabResponseCard
 from ._models_py3 import TabResponseCards
@@ -136,6 +137,7 @@
     "MeetingParticipantInfo",
     "CacheInfo",
     "TabContext",
+    "TabEntityContext",
     "TabRequest",
     "TabResponseCard",
     "TabResponseCards",