A small change to subset