diff --git a/tdb/tdb.c b/tdb/tdb.c
index 9ad6deb..7476492 100644
--- a/tdb/tdb.c
+++ b/tdb/tdb.c
@@ -35,7 +35,8 @@ void tdb_shell(tvm_t *vm)
 
 		for(i = 0; i < MAX_TOKENS; i++) memset(tokens[i], 0, TOKEN_LENGTH);
 
-		fgets(str, MAX_INPUT_LENGTH, stdin);
+		if (!fgets(str, MAX_INPUT_LENGTH, stdin))
+			break;
 		tokenize(str, (char **)tokens);
 
 		switch(cmd_to_idx(tokens[0]))