diff --git a/334/homework/2/msh.c b/334/homework/2/msh.c
index 956d3de..218efb4 100644
--- a/334/homework/2/msh.c
+++ b/334/homework/2/msh.c
@@ -15,6 +15,9 @@ int main(void) {
 	while(1) {
 		printf("%s", PROMPT);
 		fgets(buffer, MAX_BUF, stdin);
+		if(!strlen(buffer)) {
+			return 0;
+		}
 		// process the input
 		remove_newline(buffer);
 		exit_branch(buffer);