Add the line I accidentally deleted 

probably not the right way to do this, I think it will be ok