Add test for Rollback task