From 62b3d98cc19f3563101c5d057ca33193a6935095 Mon Sep 17 00:00:00 2001
From: Maschell <Maschell@gmx.de>
Date: Tue, 21 Mar 2023 22:58:56 +0100
Subject: [PATCH] For memory testing

---
 source/main.cpp | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/source/main.cpp b/source/main.cpp
index c0b7d7e..fa7cd35 100644
--- a/source/main.cpp
+++ b/source/main.cpp
@@ -156,6 +156,8 @@ WUMS_APPLICATION_STARTS() {
 
     OSReport("Running FunctionPatcherModule " MODULE_VERSION_FULL "\n");
 
+    gFunctionAddressProvider->resetHandles();
+
     // Now we can update the pointer with the "real" functions
     gMEMAllocFromDefaultHeapExForThreads = MEMAllocFromDefaultHeapEx;
     gMEMFreeToDefaultHeapForThreads      = MEMFreeToDefaultHeap;
@@ -178,9 +180,7 @@ WUMS_APPLICATION_STARTS() {
 WUMS_APPLICATION_REQUESTS_EXIT() {
     deinitLogging();
 }
-WUMS_APPLICATION_ENDS() {
-    gFunctionAddressProvider->resetHandles();
-}
+
 
 WUMS_EXPORT_FUNCTION(FunctionPatcherPatchFunction);
 WUMS_EXPORT_FUNCTION(FunctionPatcherRestoreFunction);
\ No newline at end of file