diff --git a/Makefile b/Makefile
index c7aee9b..f435219 100644
--- a/Makefile
+++ b/Makefile
@@ -11,8 +11,9 @@ test-cover:
 	go tool cover -html=coverage.out
 
 gen:
-	cd pocketic && go generate
 	cd candid && go generate
+	cd pocketic && go generate
+	cd registry && go generate
 
 gen-ic:
 	go run ic/testdata/gen.go
diff --git a/registry/proto/v1/registry.pb.go b/registry/proto/v1/registry.pb.go
index 08380d2..99341d5 100644
--- a/registry/proto/v1/registry.pb.go
+++ b/registry/proto/v1/registry.pb.go
@@ -70,56 +70,6 @@ var file_registry_proto_rawDesc = []byte{
 	0x6f, 0x2f, 0x76, 0x31, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33,
 }
 
-func file_registry_proto_init() {
-	if File_registry_proto != nil {
-		return
-	}
-	if !protoimpl.UnsafeEnabled {
-		file_registry_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
-			switch v := v.(*ProtoRegistry); i {
-			case 0:
-				return &v.state
-			case 1:
-				return &v.sizeCache
-			case 2:
-				return &v.unknownFields
-			default:
-				return nil
-			}
-		}
-		file_registry_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
-			switch v := v.(*ProtoRegistryRecord); i {
-			case 0:
-				return &v.state
-			case 1:
-				return &v.sizeCache
-			case 2:
-				return &v.unknownFields
-			default:
-				return nil
-			}
-		}
-	}
-	type x struct{}
-	out := protoimpl.TypeBuilder{
-		File: protoimpl.DescBuilder{
-			GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
-			RawDescriptor: file_registry_proto_rawDesc,
-			NumEnums:      0,
-			NumMessages:   2,
-			NumExtensions: 0,
-			NumServices:   0,
-		},
-		GoTypes:           file_registry_proto_goTypes,
-		DependencyIndexes: file_registry_proto_depIdxs,
-		MessageInfos:      file_registry_proto_msgTypes,
-	}.Build()
-	File_registry_proto = out.File
-	file_registry_proto_rawDesc = nil
-	file_registry_proto_goTypes = nil
-	file_registry_proto_depIdxs = nil
-}
-
 func file_registry_proto_rawDescGZIP() []byte {
 	file_registry_proto_rawDescOnce.Do(func() {
 		file_registry_proto_rawDescData = protoimpl.X.CompressGZIP(file_registry_proto_rawDescData)
@@ -127,8 +77,6 @@ func file_registry_proto_rawDescGZIP() []byte {
 	return file_registry_proto_rawDescData
 }
 
-func init() { file_registry_proto_init() }
-
 type ProtoRegistry struct {
 	state         protoimpl.MessageState
 	sizeCache     protoimpl.SizeCache
@@ -211,7 +159,9 @@ func (x *ProtoRegistryRecord) GetVersion() uint64 {
 	}
 	return 0
 }
+
 func (*ProtoRegistryRecord) ProtoMessage() {}
+
 func (x *ProtoRegistryRecord) ProtoReflect() protoreflect.Message {
 	mi := &file_registry_proto_msgTypes[1]
 	if protoimpl.UnsafeEnabled && x != nil {
@@ -223,7 +173,6 @@ func (x *ProtoRegistryRecord) ProtoReflect() protoreflect.Message {
 	}
 	return mi.MessageOf(x)
 }
-
 func (x *ProtoRegistryRecord) Reset() {
 	*x = ProtoRegistryRecord{}
 	if protoimpl.UnsafeEnabled {
@@ -235,3 +184,54 @@ func (x *ProtoRegistryRecord) Reset() {
 func (x *ProtoRegistryRecord) String() string {
 	return protoimpl.X.MessageStringOf(x)
 }
+
+func init() { file_registry_proto_init() }
+func file_registry_proto_init() {
+	if File_registry_proto != nil {
+		return
+	}
+	if !protoimpl.UnsafeEnabled {
+		file_registry_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
+			switch v := v.(*ProtoRegistry); i {
+			case 0:
+				return &v.state
+			case 1:
+				return &v.sizeCache
+			case 2:
+				return &v.unknownFields
+			default:
+				return nil
+			}
+		}
+		file_registry_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
+			switch v := v.(*ProtoRegistryRecord); i {
+			case 0:
+				return &v.state
+			case 1:
+				return &v.sizeCache
+			case 2:
+				return &v.unknownFields
+			default:
+				return nil
+			}
+		}
+	}
+	type x struct{}
+	out := protoimpl.TypeBuilder{
+		File: protoimpl.DescBuilder{
+			GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
+			RawDescriptor: file_registry_proto_rawDesc,
+			NumEnums:      0,
+			NumMessages:   2,
+			NumExtensions: 0,
+			NumServices:   0,
+		},
+		GoTypes:           file_registry_proto_goTypes,
+		DependencyIndexes: file_registry_proto_depIdxs,
+		MessageInfos:      file_registry_proto_msgTypes,
+	}.Build()
+	File_registry_proto = out.File
+	file_registry_proto_rawDesc = nil
+	file_registry_proto_goTypes = nil
+	file_registry_proto_depIdxs = nil
+}