The fide rules states that it is a draw when the flag falls if there are no legal series of moves that can result in the person with time remaining checkmating. At the end of http://en.lichess.org/m3dpK5v0TWEH it is still possible for white to checkmate so by fide rules this would be a win.
I'm not sure if this will fixed, since it would be hard to fully implement the rules, however the flag fall implementation could be brought closer to the fide rules in practice, so it is something worth considering.
Possible improvement:
If the player with time has a knight and the opposing player has a pawn/knight then it is a win. (I;m assuming this isn't implemented, I haven't checked)
A bit more complicated:
If the player with time has a bishop and the opposing player has the opposite color bishop, or can promote a pawn to the opposite color bishop then it is a win.