diff --git a/src/test/java/com/hivemq/cli/commands/hivemq/behaviorstate/BehaviorStateGetCommandTest.java b/src/test/java/com/hivemq/cli/commands/hivemq/behaviorstate/BehaviorStateGetCommandTest.java
index 558856bc9..ba621a2c7 100644
--- a/src/test/java/com/hivemq/cli/commands/hivemq/behaviorstate/BehaviorStateGetCommandTest.java
+++ b/src/test/java/com/hivemq/cli/commands/hivemq/behaviorstate/BehaviorStateGetCommandTest.java
@@ -38,7 +38,7 @@ public class BehaviorStateGetCommandTest {
 
     private final @NotNull HiveMQRestService hiveMQRestService = mock();
     private final @NotNull OutputFormatter outputFormatter = mock();
-    private final @NotNull DataHubStateApi stateApi = mock(DataHubStateApi.class);
+    private final @NotNull DataHubStateApi stateApi = mock();
 
     private final @NotNull CommandLine commandLine =
             new CommandLine(new BehaviorStateGetCommand(hiveMQRestService, outputFormatter));