Are you sure about that? I thought the bug persisted in SMB in Super Mario All-Stars. (Of course, the number of lives will at least display correctly.)
Anyway, you might want to take your question to a more specialized board. It's probably possible, given that there's a complete annotated disassembly of the SMB ROM out there, but I doubt it's straightforward – things probably have to be rearranged substantially.