Minor refactoring

serenity-bdd · Apr 21, 2024 · 2b6a933 · 2b6a933
1 parent afdec2c
commit 2b6a933
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/serenity-core/src/main/java/net/serenitybdd/core/sessions/TestSessionVariables.java b/serenity-core/src/main/java/net/serenitybdd/core/sessions/TestSessionVariables.java
@@ -6,9 +6,9 @@
 import java.util.Map;
 import java.util.concurrent.ConcurrentHashMap;
 
-public class TestSessionVariables extends ConcurrentHashMap implements SessionMap {
+public class TestSessionVariables<K,V> extends ConcurrentHashMap<K,V> implements SessionMap<K,V> {
 
-    private final Map<String, String> metadata = new ConcurrentHashMap();
+    private final Map<String, String> metadata = new ConcurrentHashMap<>();
 
     @Override
     public void shouldContainKey(Object key) {
@@ -20,7 +20,7 @@ public void shouldContainKey(Object key) {
 
 
     @Override
-    public Object put(Object key, Object value) {
+    public V put(K key, V value) {
         if (value == null) {
             return remove(key);
         } else {