Fix mongo_setup import