Fixed lasso bug