Fix go test issue on sub packages