diff --git a/Source/HBIOS/hbios.asm b/Source/HBIOS/hbios.asm
index 114f7001..eecebdfb 100644
--- a/Source/HBIOS/hbios.asm
+++ b/Source/HBIOS/hbios.asm
@@ -5284,6 +5284,7 @@ SYS_GETDIOMED:
 ;
 	; CHECK MEDIA TYPE, ONLY HD IS APPLICABLE
 	LD	A,E			; RESULTANT MEDIA ID TO ACCUM
+	LD	(DIOMED_MID),A		; INIT VALUE, MAY BE USED LATER
 	OR	A			; SET FLAGS
 	JR	Z, SYS_GETDIOMED1A	; BAIL IF NO MEDIA
 	CP	MID_HD			; IS IT A HARD DISK
@@ -5296,6 +5297,9 @@ SYS_GETDIOMED:
 ;
 SYS_GETDIOMED1A:
 	; RETURN MEDIA ID (NOT HD) WITH SUCCESS
+	LD	A,(DIOMED_DEVATT)	; GET DEVICE ATTRIBUTES
+	LD	B,A			; ... AND RETURN IN B
+	LD	C,E			; RETURN MEDIA ID IN C
 	LD	DE,0			; LBA VALUE WILL BE ZERO
 	LD	HL,0
 	JP	SYS_GETDIOMED6A		; RETURN SUCCESS