Fix tests

rbertoche · Aug 13, 2023 · bee00b1 · bee00b1
1 parent e76d037
commit bee00b1
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/Content.IntegrationTests/Tests/Disease/TryAddDisease.cs b/Content.IntegrationTests/Tests/Disease/TryAddDisease.cs
@@ -1,7 +1,5 @@
-using System.Threading.Tasks;
 using Content.Server.Disease;
 using Content.Shared.Disease;
-using NUnit.Framework;
 using Robust.Shared.GameObjects;
 using Robust.Shared.IoC;
 using Robust.Shared.Prototypes;
@@ -31,5 +29,7 @@ await server.WaitPost(() =>
                 diseaseSystem.TryAddDisease(sickEntity, diseaseProto);
             }
         });
+
+        await pairTracker.CleanReturnAsync();
     }
 }
diff --git a/Content.IntegrationTests/Tests/Nyanotrasen/ReverseEngineering/ReverseEngineeringTest.cs b/Content.IntegrationTests/Tests/Nyanotrasen/ReverseEngineering/ReverseEngineeringTest.cs
@@ -1,7 +1,5 @@
 #nullable enable
 using System.Collections.Generic;
-using NUnit.Framework;
-using System.Threading.Tasks;
 using Content.Shared.ReverseEngineering;
 using Content.Shared.Research.Prototypes;
 using Content.Server.ReverseEngineering;
@@ -66,6 +64,8 @@ await server.WaitAssertion(() =>
                     }
                 }
             });
+
+            await pairTracker.CleanReturnAsync();
         }
     }
 }