whichbas: more set_type().

1 files changed, 4 insertions, 4 deletions

diff --git a/whichbas.c b/whichbas.c
index ddea7a7..49b27c9 100644
--- a/whichbas.c
+++ b/whichbas.c
@@ -174,12 +174,12 @@ CALLBACK(handle_cmd) {
 								"pos %04x, pos_after_string %04x (token %02x)\n",
 								lineno, pos, pos_after_string, program[pos_after_string]);
 					if(program[pos_after_string] == OP_SEMICOLON) {
-						remove_type(BT_BXL_BXE);
+						set_type(BT_TURBO);
 					}
 				}
 			} else { /* has_args is false, oh shit! */
 				fprintf(stderr, "*** INPUT without variable at line %d.\n*** Rev A BASIC bug, program will crash, better fix it!\n", lineno);
-				remove_type(BT_TURBO | BT_BXL_BXE);
+				set_type(BT_ATARI);
 			}
 			break;
 		case CMD_GET:
@@ -280,7 +280,7 @@ CALLBACK(handle_cmd) {
             BXE: PROCEDURE (arg is string const (not var!)) */
 			if(nexttok == OP_STRCONST) {
 				/* this token doesn't seem to be valid in BXL at all */
-				remove_type(BT_TURBO | BT_BXL);
+				set_type(BT_BXE);
 			} else {
 				remove_type(BT_BXL_BXE);
 			}
@@ -631,7 +631,7 @@ CALLBACK(handle_end_stmt) {
 			   Turbo BGET and BPUT always take 3 args. So does BXL BPUT.
 				BXE BPUT takes 3 args and an optional 4th. */
 			if(comma_count != 2) {
-				remove_type(BT_TURBO | BT_BXL);
+				set_type(BT_BXE);
 			}
 			break;
       case 0x55: /* RENUM in both (TB req 3 args, BXL up to two) */