Add a spec for this?