Restrict now returns a factor