Small change to accommodate missing data