diff --git a/proto/manager_protobuf.proto b/proto/manager_protobuf.proto
index fc5bb24..217dadb 100644
--- a/proto/manager_protobuf.proto
+++ b/proto/manager_protobuf.proto
@@ -33,9 +33,9 @@ message EnvironmentEntry {
 message RestartPolicy {
   uint32 max_retries = 1;
   uint32 delay_seconds = 2;
-  oneof policy {
-    bool ALWAYS = 3;
-    bool ON_NON_ZERO_EXIT = 4;
+  oneof policy_type_type {
+    bool Always = 3;
+    bool OnNonZeroExit = 4;
   }
 }
 
@@ -43,7 +43,7 @@ message RestartPolicy {
 message ChildProcess {
   string path = 1;
   repeated string arguments = 2;
-  RestartPolicy restart_policy = 3;
+  RestartPolicy restart= 3;
 }
 
 service ConfigManager {
diff --git a/src/lib.rs b/src/lib.rs
index 25c7ee2..f988b12 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -58,7 +58,7 @@ pub mod config {
         #[prost(string, repeated, tag = "2")]
         pub arguments: Vec<String>,
         #[prost(message, optional, tag = "3")]
-        restart_policy: Option<RestartPolicy>,
+        restart: Option<RestartPolicy>,
     }
 
     #[derive(Clone, Serialize, Deserialize, prost::Message)]