Sorry seems like you lazy loaded some part after but not the edit_mode thing.