I've just got an LED 501 in there - but make sure it's a stubby one as the longer ones, you won't get the cover back on.
However - if you do find a replacement unit let me know what you go for - as the same as you, I've always replaced the complete unit to mega bright LED ones in the past