Unit test fixes

[Validator][Debug] Fixing unit tests

| Q             | A
| ------------- | ---
| Bug fix?      | yes
| New feature?  | no
| BC breaks?    | no
| Deprecations? | no
| Tests pass?   | yes
| Fixed tickets | 
| License       | MIT
| Doc PR        | 
couldn't it be restored before the assertions, so that it is also restored in case of test failure ?