Adding tests for the rollback cmd