Fix + golf

The function didn't properly.
Test case : it was impossible to run it twice in a row.