diff --git a/LDK/src/org/labkey/ldk/LDKModule.java b/LDK/src/org/labkey/ldk/LDKModule.java
index 78b1a83b..ddbccaf2 100644
--- a/LDK/src/org/labkey/ldk/LDKModule.java
+++ b/LDK/src/org/labkey/ldk/LDKModule.java
@@ -150,7 +150,7 @@ public static boolean isSqlServer()
     }
 
     @Override
-    public @NotNull Set<Class> getIntegrationTests()
+    public @NotNull Set<Class<?>> getIntegrationTests()
     {
         return PageFlowUtil.set(LookupSetTable.TestCase.class);
     }
diff --git a/laboratory/src/org/labkey/laboratory/LaboratoryModule.java b/laboratory/src/org/labkey/laboratory/LaboratoryModule.java
index 6cb8a7d8..4a462b16 100644
--- a/laboratory/src/org/labkey/laboratory/LaboratoryModule.java
+++ b/laboratory/src/org/labkey/laboratory/LaboratoryModule.java
@@ -235,7 +235,7 @@ public UpgradeCode getUpgradeCode()
     }
 
     @Override
-    public @NotNull Set<Class> getIntegrationTests()
+    public @NotNull Set<Class<?>> getIntegrationTests()
     {
         return PageFlowUtil.set(WorkbookTestCase.class);
     }