diff --git a/unmarshalJSON.go b/unmarshalJSON.go
index d8802fa44b7ad1eb9f5985fa8d3734713ea3a27f..1af792b34bcb894c7d7ef64f5fd702f645be52a2 100644
--- a/unmarshalJSON.go
+++ b/unmarshalJSON.go
@@ -8,7 +8,7 @@ package query
 import (
 	"encoding/json"
 
-	"git.science.uu.nl/graphpolaris/query-conversion/entity"
+	"git.science.uu.nl/graphpolaris/query-conversion/entityv2"
 )
 
 /*
@@ -16,8 +16,8 @@ UnmarshalJSON takes an incoming message and unmarshals it
 	msg: *[]byte, the incoming message
 	Returns: *entity.IncomingQueryJSON, the unmarshalled message and a possible error
 */
-func UnmarshalJSON(msg *[]byte) (*entity.IncomingQueryJSON, error) {
-	var JSONQuery entity.IncomingQueryJSON
+func UnmarshalJSON(msg *[]byte) (*entityv2.IncomingQueryJSON, error) {
+	var JSONQuery entityv2.IncomingQueryJSON
 	err := json.Unmarshal(*msg, &JSONQuery)
 	return &JSONQuery, err
 }