1 files changed, 8 insertions, 7 deletions
diff --git a/Swiften/Network/PlatformNATTraversalWorker.h b/Swiften/Network/PlatformNATTraversalWorker.h
index 7c249cc..9de1258 100644
--- a/Swiften/Network/PlatformNATTraversalWorker.h
+++ b/Swiften/Network/PlatformNATTraversalWorker.h
@@ -12,17 +12,18 @@
 #include <boost/thread/mutex.hpp>
 #include <boost/thread/condition_variable.hpp>
 
+#include <Swiften/Network/NATTraverser.h>
 #include <Swiften/Network/HostAddressPort.h>
 #include <Swiften/Network/PlatformNATTraversalRequest.h>
 
 namespace Swift {
 
 class EventLoop;
-class PlatformNATTraversalGetPublicIPRequest;
-class PlatformNATTraversalForwardPortRequest;
-class PlatformNATTraversalRemovePortForwardingRequest;
+class NATTraversalGetPublicIPRequest;
+class NATTraversalForwardPortRequest;
+class NATTraversalRemovePortForwardingRequest;
 
-class PlatformNATTraversalWorker {
+class PlatformNATTraversalWorker : public NATTraverser {
 private:
 	enum BackendType {
 		NotYetDecided,
@@ -35,9 +36,9 @@ public:
 	PlatformNATTraversalWorker(EventLoop* eventLoop);
 	~PlatformNATTraversalWorker();
 
-	boost::shared_ptr<PlatformNATTraversalGetPublicIPRequest> createGetPublicIPRequest();
-	boost::shared_ptr<PlatformNATTraversalForwardPortRequest> createForwardPortRequest(unsigned int localPort, unsigned int publicPort);
-	boost::shared_ptr<PlatformNATTraversalRemovePortForwardingRequest> createRemovePortForwardingRequest(unsigned int localPort, unsigned int publicPort);
+	boost::shared_ptr<NATTraversalGetPublicIPRequest> createGetPublicIPRequest();
+	boost::shared_ptr<NATTraversalForwardPortRequest> createForwardPortRequest(unsigned int localPort, unsigned int publicPort);
+	boost::shared_ptr<NATTraversalRemovePortForwardingRequest> createRemovePortForwardingRequest(unsigned int localPort, unsigned int publicPort);
 
 	void run();
 	void addRequestToQueue(PlatformNATTraversalRequest::ref);