From 3a722acc629759737425d2be31bef2f178239fd0 Mon Sep 17 00:00:00 2001
From: Damian Korczowski <erdroy2@gmail.com>
Date: Mon, 21 Jun 2021 20:32:02 +0200
Subject: [PATCH] Cleanup

---
 Source/Engine/Networking/Drivers/ENetDriver.h | 2 --
 Source/Engine/Networking/INetworkDriver.h     | 3 +--
 Source/Engine/Networking/NetworkManager.h     | 2 --
 Source/Engine/Networking/Types.h              | 1 +
 4 files changed, 2 insertions(+), 6 deletions(-)

diff --git a/Source/Engine/Networking/Drivers/ENetDriver.h b/Source/Engine/Networking/Drivers/ENetDriver.h
index 70b97a74d..86e4635e0 100644
--- a/Source/Engine/Networking/Drivers/ENetDriver.h
+++ b/Source/Engine/Networking/Drivers/ENetDriver.h
@@ -10,8 +10,6 @@
 #include "Engine/Core/Collections/Dictionary.h"
 #include "Engine/Scripting/ScriptingType.h"
 
-class NetworkPeer;
-
 API_CLASS(Namespace="FlaxEngine.Networking", Sealed) class FLAXENGINE_API ENetDriver : public INetworkDriver
 {
 DECLARE_SCRIPTING_TYPE_MINIMAL(ENetDriver);
diff --git a/Source/Engine/Networking/INetworkDriver.h b/Source/Engine/Networking/INetworkDriver.h
index 8d617e10f..bcdf65079 100644
--- a/Source/Engine/Networking/INetworkDriver.h
+++ b/Source/Engine/Networking/INetworkDriver.h
@@ -3,8 +3,7 @@
 #pragma once
 
 #include "Engine/Scripting/ScriptingType.h"
-
-class NetworkPeer;
+#include "Types.h"
 
 API_INTERFACE(Namespace="FlaxEngine.Networking") class FLAXENGINE_API INetworkDriver
 {
diff --git a/Source/Engine/Networking/NetworkManager.h b/Source/Engine/Networking/NetworkManager.h
index 723ba0985..16e0f73b9 100644
--- a/Source/Engine/Networking/NetworkManager.h
+++ b/Source/Engine/Networking/NetworkManager.h
@@ -5,8 +5,6 @@
 #include "Engine/Scripting/ScriptingType.h"
 #include "Types.h"
 
-class NetworkPeer;
-
 API_CLASS(Namespace="FlaxEngine.Networking", Static) class FLAXENGINE_API NetworkManager
 {
 DECLARE_SCRIPTING_TYPE_NO_SPAWN(NetworkManager);
diff --git a/Source/Engine/Networking/Types.h b/Source/Engine/Networking/Types.h
index c92eabac8..7f3a157d1 100644
--- a/Source/Engine/Networking/Types.h
+++ b/Source/Engine/Networking/Types.h
@@ -6,6 +6,7 @@ enum class NetworkChannelType;
 enum class NetworkEventType;
 
 class INetworkDriver;
+class NetworkPeer;
 
 struct NetworkEvent;
 struct NetworkConnection;