Fix blocked lasso notebook from JT's hint