Fix Save register

Actually mainly python script bug