diff --git a/client/src/main/java/com/vesoft/nebula/client/graph/net/Session.java b/client/src/main/java/com/vesoft/nebula/client/graph/net/Session.java
index 2a17fa88..f68f6a86 100644
--- a/client/src/main/java/com/vesoft/nebula/client/graph/net/Session.java
+++ b/client/src/main/java/com/vesoft/nebula/client/graph/net/Session.java
@@ -532,7 +532,7 @@ public synchronized void close() {
             put(Boolean.class, (Setter<Boolean>) Value::bVal);
             put(Integer.class, (Setter<Integer>) Value::iVal);
             put(Short.class, (Setter<Short>) Value::iVal);
-            put(Byte.class, (Setter<Short>) Value::iVal);
+            put(Byte.class, (Setter<Byte>) Value::iVal);
             put(Long.class, (Setter<Long>) Value::iVal);
             put(Float.class, (Setter<Float>) Value::fVal);
             put(Double.class, (Setter<Double>) Value::fVal);