diff --git a/TRDOS/TR_HDD_4.ASZ b/TRDOS/TR_HDD_4.ASZ
index a08b573..baa5a77 100644
--- a/TRDOS/TR_HDD_4.ASZ
+++ b/TRDOS/TR_HDD_4.ASZ
@@ -82,6 +82,7 @@ HDD_SET:
 		JR	C,.init
 		;
 		LD	(SYS_PAGE.CURRENT_DIR_DRIVE),A
+		LD	E,A
 		OUT	(C),B
 		PUSH	DE
 		AND	#0F
@@ -96,7 +97,7 @@ HDD_SET:
 		LD	C,BIOS.HDD_INIT
 		CALL	EXP_FNS
 		POP	DE
-		JP	C,HDD_PROG_E
+		JR	C,HDD_PROG_E
 		;
 .exit:		LD	C,BIOS.SET_DISK_REDIR
 		JP	EXP_FNS