Fix unit tests for reset