Fixed column management

This fixes columns not showing up in the stage edit view, which was causing some weird behavior.