Move it into tests