1 files changed, 7 insertions, 8 deletions
diff --git a/src/components/transport_manager/include/transport_manager/transport_manager_default.h b/src/components/transport_manager/include/transport_manager/transport_manager_default.h
index 63061b14b7..9d428cf817 100644
--- a/src/components/transport_manager/include/transport_manager/transport_manager_default.h
+++ b/src/components/transport_manager/include/transport_manager/transport_manager_default.h
@@ -38,10 +38,6 @@
 
 #include "transport_manager/transport_manager_impl.h"
 
-namespace resumption {
-class LastState;
-}
-
 namespace transport_manager {
 
 struct TransportAdapterFactory {
@@ -50,19 +46,19 @@ struct TransportAdapterFactory {
   using CreatorTA =
       std::function<transport_adapter::TransportAdapter*(Args&&... args)>;
 #ifdef BLUETOOTH_SUPPORT
-  CreatorTA<resumption::LastState&, const TransportManagerSettings&>
+  CreatorTA<resumption::LastStateWrapperPtr&, const TransportManagerSettings&>
       ta_bluetooth_creator_;
 #endif
   CreatorTA<const uint16_t,
-            resumption::LastState&,
+            resumption::LastStateWrapperPtr&,
             const TransportManagerSettings&>
       ta_tcp_creator_;
 #if defined(USB_SUPPORT)
-  CreatorTA<resumption::LastState&, const TransportManagerSettings&>
+  CreatorTA<resumption::LastStateWrapperPtr&, const TransportManagerSettings&>
       ta_usb_creator_;
 #endif
 #if defined(CLOUD_APP_WEBSOCKET_TRANSPORT_SUPPORT)
-  CreatorTA<resumption::LastState&, const TransportManagerSettings&>
+  CreatorTA<resumption::LastStateWrapperPtr&, const TransportManagerSettings&>
       ta_cloud_creator_;
 #endif
 };
@@ -80,6 +76,9 @@ class TransportManagerDefault : public TransportManagerImpl {
    *
    * @return Code error.
    */
+  int Init(resumption::LastStateWrapperPtr last_state_wrapper) OVERRIDE;
+
+  DEPRECATED
   int Init(resumption::LastState& last_state) OVERRIDE;
 
   /**