Imports do not work

[kotyara85]

Hey there,
I couldn't make import work property with graphql

Here's import example, both files are in the same directory:
a.proto

package a.v1;

option go_package = "example.com/repo/gen/v1/a;v1";

import "b.proto";

message IndexResponse {
  repeated b.v1.Test clusters = 1;
}

b.proto

package b.v1;

option go_package = "example.com/repo/gen/v1/b;v1";

message Test {
  string name = 1;
}

Generated file has a reference but not using them so in a package it will look like this

Type: graphql.NewList(Gql__type_Test()), instead of v1.Gql__type_Test()

Any help is appreciated