Just use LDX $BA instead of LDX #$08 before JSR $FFBA in floppy routines to make them work from drive number you loaded from.