diff --git a/query/gremlin/session.go b/query/gremlin/session.go
index e5b3460..3f8bb0b 100644
--- a/query/gremlin/session.go
+++ b/query/gremlin/session.go
@@ -47,9 +47,9 @@ type Session struct {
 	emptyEnv   *otto.Otto
 }
 
-func NewSession(inputTripleStore graph.TripleStore, timeoutSec int, persist bool) *Session {
+func NewSession(ts graph.TripleStore, timeoutSec int, persist bool) *Session {
 	g := Session{
-		ts:         inputTripleStore,
+		ts:         ts,
 		limit:      -1,
 		timeoutSec: time.Duration(timeoutSec),
 	}