diff --git a/include/entitycache.hpp b/include/entitycache.hpp
index 01d437f5..fcb613cc 100644
--- a/include/entitycache.hpp
+++ b/include/entitycache.hpp
@@ -22,6 +22,8 @@
 #include "playerresource.h"
 #include "globals.h"
 #include "classinfo.hpp"
+#include "tf2.gen.hpp"
+#include "client_class.h"
 
 struct matrix3x4_t;
 
diff --git a/src/entitycache.cpp b/src/entitycache.cpp
index f4cc6b03..4383085a 100644
--- a/src/entitycache.cpp
+++ b/src/entitycache.cpp
@@ -139,8 +139,6 @@ CachedEntity array[MAX_ENTITIES]{};
 
 void Update()
 {
-	PROF_SECTION(CACHE_UPDATE)
-		{
     max = g_IEntityList->GetHighestEntityIndex();
     if (max >= MAX_ENTITIES)
         max = MAX_ENTITIES - 1;
@@ -148,7 +146,6 @@ void Update()
     {
         array[i].Update();
     }
-		}
 }
 
 void Invalidate()