Can we remove the TODO and also add the testing of the following scenarios:
1. Execution within a transaction
2. Restart (the reset of the count)