diff --git a/src/carnot/goplanner/schema_test.go b/src/carnot/goplanner/schema_test.go
index db4e4df2295..f1eea4b69a2 100644
--- a/src/carnot/goplanner/schema_test.go
+++ b/src/carnot/goplanner/schema_test.go
@@ -44,6 +44,16 @@ relation_map {
 			column_type: INT64
 			column_semantic_type: ST_PORT
 		}
+		columns {
+			column_name: "local_addr"
+			column_type: STRING
+			column_semantic_type: ST_IP_ADDRESS
+		}
+		columns {
+			column_name: "local_port"
+			column_type: INT64
+			column_semantic_type: ST_PORT
+		}
 		columns {
 			column_name: "protocol"
 			column_type: INT64