Fix bug in lasso selection