Add missing imports in mongo_transaction