shouldn't have to `.flatten` here.