Make sure rollbacks state gets restored

When  config is set to raise, we need still to make sure the records are properly restored