diff --git a/aiserver.py b/aiserver.py index 0608e50b..b6940e1e 100644 --- a/aiserver.py +++ b/aiserver.py @@ -422,7 +422,7 @@ class ImportBuffer: status = err.status_code print(f"[import] Got {status} on request to club :^(") message = f"Club responded with {status}" - if status == "404": + if status == 404: message = f"Prompt not found for ID {club_id}" show_error_notification("Error loading prompt", message) return diff --git a/importers/aetherroom.py b/importers/aetherroom.py index e27d1439..33ea37f5 100644 --- a/importers/aetherroom.py +++ b/importers/aetherroom.py @@ -16,7 +16,7 @@ class ImportData: class RequestFailed(Exception): - def __init__(self, status_code: str) -> None: + def __init__(self, status_code: int) -> None: self.status_code = status_code super().__init__()