1 files changed, 5 insertions, 2 deletions
diff --git a/src/base/QXmppStreamManagement_p.h b/src/base/QXmppStreamManagement_p.h
index fb79c7b0..1a528c0c 100644
--- a/src/base/QXmppStreamManagement_p.h
+++ b/src/base/QXmppStreamManagement_p.h
@@ -31,6 +31,7 @@
 #include <QXmlStreamWriter>
 
 class QXmppStream;
+class QXmppPacket;
 
 //
 //  W A R N I N G
@@ -189,12 +190,14 @@ class QXmppStreamManager
 {
 public:
     explicit QXmppStreamManager(QXmppStream *stream);
+    ~QXmppStreamManager();
 
+    bool enabled() const;
     unsigned int lastIncomingSequenceNumber() const;
 
     void handleDisconnect();
     void handleStart();
-    void handlePacketSent(const QXmppStanza &packet, const QByteArray &data);
+    void handlePacketSent(QXmppPacket &packet);
     bool handleStanza(const QDomElement &stanza);
 
     void resetCache();
@@ -210,7 +213,7 @@ private:
     QXmppStream *stream;
 
     bool m_enabled = false;
-    QMap<unsigned int, QByteArray> m_unacknowledgedStanzas;
+    QMap<unsigned int, QXmppPacket> m_unacknowledgedStanzas;
     unsigned int m_lastOutgoingSequenceNumber = 0;
     unsigned int m_lastIncomingSequenceNumber = 0;
 };