diff --git a/packages/stream_chat/lib/stream_chat.dart b/packages/stream_chat/lib/stream_chat.dart
index 0479a3786..4c4edf9e6 100644
--- a/packages/stream_chat/lib/stream_chat.dart
+++ b/packages/stream_chat/lib/stream_chat.dart
@@ -66,4 +66,5 @@ export 'src/core/models/user.dart';
 export 'src/core/util/extension.dart';
 export 'src/db/chat_persistence_client.dart';
 export 'src/event_type.dart';
+export 'src/stream_package.dart';
 export 'src/ws/connection_status.dart';
diff --git a/packages/stream_chat_flutter/lib/src/stream_chat.dart b/packages/stream_chat_flutter/lib/src/stream_chat.dart
index 287f08a08..9ff0082ee 100644
--- a/packages/stream_chat_flutter/lib/src/stream_chat.dart
+++ b/packages/stream_chat_flutter/lib/src/stream_chat.dart
@@ -108,6 +108,7 @@ class StreamChatState extends State<StreamChat> {
                 connectivityStream: widget.connectivityStream,
                 child: Builder(
                   builder: (context) {
+                    StreamChatClient.packageInUse = StreamPackage.ui;
                     return widget.child ?? const Offstage();
                   },
                 ),
diff --git a/packages/stream_chat_flutter_core/lib/src/stream_chat_core.dart b/packages/stream_chat_flutter_core/lib/src/stream_chat_core.dart
index 12234a5e6..f428fd570 100644
--- a/packages/stream_chat_flutter_core/lib/src/stream_chat_core.dart
+++ b/packages/stream_chat_flutter_core/lib/src/stream_chat_core.dart
@@ -95,6 +95,7 @@ class StreamChatCoreState extends State<StreamChatCore>
 
   @override
   Widget build(BuildContext context) {
+    StreamChatClient.packageInUse = StreamPackage.core;
     return widget.child;
   }