diff --git a/src/DelegateDecompiler.EntityFramework.Tests/TestGroup20Relationship/Test01Include.cs b/src/DelegateDecompiler.EntityFramework.Tests/TestGroup20Relationship/Test01Include.cs
index ffe2a995..d999822c 100644
--- a/src/DelegateDecompiler.EntityFramework.Tests/TestGroup20Relationship/Test01Include.cs
+++ b/src/DelegateDecompiler.EntityFramework.Tests/TestGroup20Relationship/Test01Include.cs
@@ -4,8 +4,11 @@
 using DelegateDecompiler.EntityFramework.Tests.Helpers;
 using NUnit.Framework;
 using DelegateDecompiler.EntityFramework.Tests.EfItems;
+#if EF_CORE
+using Microsoft.EntityFrameworkCore;
+#else 
 using System.Data.Entity;
-using System.Collections.Generic;
+#endif
 
 namespace DelegateDecompiler.EntityFramework.Tests.TestGroup05BasicFeatures
 {