diff --git a/redbot/setup.py b/redbot/setup.py index 05d249867..be6b75d8a 100644 --- a/redbot/setup.py +++ b/redbot/setup.py @@ -161,10 +161,11 @@ def basic_setup(): storage = get_storage_type() - storage_dict = {1: "JSON", 2: "MongoDB"} - default_dirs["STORAGE_TYPE"] = storage_dict.get(storage, 1) + storage_dict = {1: BackendType.JSON, 2: BackendType.MONGO} + storage_type: BackendType = storage_dict.get(storage, BackendType.JSON) + default_dirs["STORAGE_TYPE"] = storage_type.value - if storage_dict.get(storage, 1) == "MongoDB": + if storage_type == BackendType.MONGO: from redbot.core.drivers.red_mongo import get_config_details default_dirs["STORAGE_DETAILS"] = get_config_details()