diff --git a/src/ast/mod.rs b/src/ast/mod.rs
index d3a04b6..f782f1f 100644
--- a/src/ast/mod.rs
+++ b/src/ast/mod.rs
@@ -2376,6 +2376,8 @@ pub enum PeerType {
     SQLServer,
     Kafka,
     Eventhubs,
+    PubSub,
+    Elasticsearch,
 }
 
 impl fmt::Display for PeerType {
@@ -2389,6 +2391,8 @@ impl fmt::Display for PeerType {
             PeerType::SQLServer => write!(f, "SQLSERVER"),
             PeerType::Kafka => write!(f, "KAFKA"),
             PeerType::Eventhubs => write!(f, "EVENTHUBS"),
+            PeerType::PubSub => write!(f, "PUBSUB"),
+            PeerType::Elasticsearch => write!(f, "ELASTICSEARCH"),
         }
     }
 }