diff --git a/src/vty/command.c b/src/vty/command.c
index 66cf6a8..689609c 100644
--- a/src/vty/command.c
+++ b/src/vty/command.c
@@ -514,7 +514,7 @@
 		}
 	}
 
-	out = talloc_size(NULL, len + 1);
+	out = talloc_size(tall_vty_cmd_ctx, len + 1);
 	if (!out)
 		return NULL;
 
@@ -574,7 +574,7 @@
 		}
 	}
 
-	out = talloc_size(NULL, len + 1);
+	out = talloc_size(tall_vty_cmd_ctx, len + 1);
 	if (!out)
 		return NULL;
 
@@ -1317,7 +1317,7 @@
 		return NULL;
 
 	/* tmp will hold a string of len-2 chars, so 'len' size is fine */
-	tmp = talloc_size(NULL, len);
+	tmp = talloc_size(tall_vty_cmd_ctx, len);
 
 	memcpy(tmp, (str + 1), len - 2);
 	tmp[len - 2] = '\0';