Clean nans when limiting number of features