Fix test

I really have to pay more attention so "save" and when it has to be called