diff --git a/harden/config_file.py b/harden/config_file.py index 4f2e5e2..8b45432 100644 --- a/harden/config_file.py +++ b/harden/config_file.py @@ -43,9 +43,11 @@ def get_profiles(): init_config_dir() return [] - profiles = os.listdir(PROFILE_DIR) - for i in range(len(profiles)): - profiles[i] = profiles[i].replace("_config.toml", "") + all_files = os.listdir(PROFILE_DIR) + profiles = [] + for file in all_files: + if file.endswith("_config.toml"): + profiles.append(file.replace("_config.toml", "")) return profiles