Fixed problem with tests generated by generator function/methods not rollbacking/flushing database when they should.