Yeah that made no sense. In the freak chance that the bails lodge back on, the batsman's not out.
This is my reference for the rule:
Law 28 (The wicket is down) - Laws - Laws of Cricket - Laws & Spirit - Lord's
1. Wicket put down
(a) The wicket is put down if a bail is completely removed from the top of the stumps, or a stump is struck out of the ground,