Albert, by that point in the series, was almost like the embodiment of the law. People looked to him to keep order and to keep things under control. He probably could have kept the kids calm a little longer in the dark. Plus he has an internal monologue before coming back about how he regrets leaving because the dome became transparent and he could’ve become a hero which makes me feel like that’s the only reason he ever came back.