diff --git a/src/test/java/io/eigr/spawn/AbstractContainerBaseTest.java b/src/test/java/io/eigr/spawn/AbstractContainerBaseTest.java
index a753994..a6e9928 100644
--- a/src/test/java/io/eigr/spawn/AbstractContainerBaseTest.java
+++ b/src/test/java/io/eigr/spawn/AbstractContainerBaseTest.java
@@ -11,9 +11,6 @@
 import io.eigr.spawn.test.actors.JoeActor;
 import io.eigr.spawn.test.actors.StatelessNamedActor;
 import io.eigr.spawn.test.actors.UnNamedActor;
-import org.junit.After;
-import org.junit.AfterClass;
-import org.junit.Before;
 import org.junit.jupiter.api.AfterAll;
 import org.junit.jupiter.api.BeforeAll;
 import org.slf4j.Logger;
diff --git a/src/test/java/io/eigr/spawn/test/actors/StatelessNamedActor.java b/src/test/java/io/eigr/spawn/test/actors/StatelessNamedActor.java
index 3d69bd7..65fc2af 100644
--- a/src/test/java/io/eigr/spawn/test/actors/StatelessNamedActor.java
+++ b/src/test/java/io/eigr/spawn/test/actors/StatelessNamedActor.java
@@ -9,7 +9,6 @@
 import io.eigr.spawn.internal.ActionBindings;
 import io.eigr.spawn.java.test.domain.Actor.Reply;
 import io.eigr.spawn.java.test.domain.Actor.Request;
-import io.eigr.spawn.java.test.domain.Actor.State;
 
 import static io.eigr.spawn.api.actors.behaviors.ActorBehavior.action;
 import static io.eigr.spawn.api.actors.behaviors.ActorBehavior.name;
@@ -24,10 +23,7 @@ public ActorBehavior configure(BehaviorCtx context) {
         );
     }
 
-    private Value setLanguage(ActorContext<State> context, Request msg) {
-        if (context.getState().isPresent()) {
-        }
-
+    private Value setLanguage(ActorContext<?> context, Request msg) {
         return Value.at()
                 .response(Reply.newBuilder()
                         .setResponse(String.format("Hi %s. Hello From Java", msg.getLanguage()))
diff --git a/src/test/java/io/eigr/spawn/test/actors/UnNamedActor.java b/src/test/java/io/eigr/spawn/test/actors/UnNamedActor.java
index c304638..66e2bd3 100644
--- a/src/test/java/io/eigr/spawn/test/actors/UnNamedActor.java
+++ b/src/test/java/io/eigr/spawn/test/actors/UnNamedActor.java
@@ -25,9 +25,6 @@ public ActorBehavior configure(BehaviorCtx context) {
     }
 
     private Value setLanguage(ActorContext<State> context, Request msg) {
-        if (context.getState().isPresent()) {
-        }
-
         return Value.at()
                 .response(Reply.newBuilder()
                         .setResponse(String.format("Hi %s. Hello From Java", msg.getLanguage()))