2 files changed, 2 insertions, 2 deletions
diff --git a/src/main/java/au/id/zancanaro/javacheck/state/CommandListGenerator.java b/src/main/java/au/id/zancanaro/javacheck/state/CommandListGenerator.java
index 0eb1793..0aa693d 100644
--- a/src/main/java/au/id/zancanaro/javacheck/state/CommandListGenerator.java
+++ b/src/main/java/au/id/zancanaro/javacheck/state/CommandListGenerator.java
@@ -45,7 +45,7 @@ public class CommandListGenerator<S> implements Generator<CommandList<S>> {
     public ShrinkTree<CommandList<S>> generate(Random random, int size) {
         int count = random.nextInt(size);
         @SuppressWarnings("unchecked")
-        ShrinkTree<GeneratedCommand<S, ?, ?>>[] commandTrees = (ShrinkTree<GeneratedCommand<S, ?, ?>>[]) new ShrinkTree[count];
+        ShrinkTree<GeneratedCommand<S, ?, ?>>[] commandTrees = (ShrinkTree<GeneratedCommand<S, ?, ?>>[]) new ShrinkTree<?>[count];
         S state = initialState;
         for (int i = 0; i < count; ++i) {
             commandTrees[i] = commandGenerator(i, state).generate(random, size);
diff --git a/src/main/java/au/id/zancanaro/javacheck/state/GeneratedCommand.java b/src/main/java/au/id/zancanaro/javacheck/state/GeneratedCommand.java
index 85f6fdd..6da9bf2 100644
--- a/src/main/java/au/id/zancanaro/javacheck/state/GeneratedCommand.java
+++ b/src/main/java/au/id/zancanaro/javacheck/state/GeneratedCommand.java
@@ -28,7 +28,7 @@ public class GeneratedCommand<S, A, R> {
         if (this == o) return true;
         if (o == null || getClass() != o.getClass()) return false;
 
-        GeneratedCommand that = (GeneratedCommand) o;
+        GeneratedCommand<?,?,?> that = (GeneratedCommand<?,?,?>) o;
 
         return !(args != null ? !args.equals(that.args) : that.args != null)
                 && !(command != null ? !command.equals(that.command) : that.command != null);