Fix plot in notebook when t is not 1