diff --git a/src/confluent_kafka/cimpl.pyi b/src/confluent_kafka/cimpl.pyi
index ddcb91cab..f732822cf 100644
--- a/src/confluent_kafka/cimpl.pyi
+++ b/src/confluent_kafka/cimpl.pyi
@@ -226,7 +226,7 @@ class KafkaError:
     _WAIT_COORD: int
     def __init__(
         self,
-        code: int,
+        error: int,
         reason: Optional[str] = None,
         fatal: bool = False,
         retriable: bool = False,
diff --git a/tests/test_kafka_error.py b/tests/test_kafka_error.py
index 83b463767..27142f2a9 100644
--- a/tests/test_kafka_error.py
+++ b/tests/test_kafka_error.py
@@ -94,6 +94,14 @@ def test_kafkaError_custom_msg():
     assert not err.txn_requires_abort()
 
 
+def test_kafkaError_init_keyword_matches_runtime_signature():
+    err = KafkaError(error=KafkaError._ALL_BROKERS_DOWN)
+    assert err == KafkaError._ALL_BROKERS_DOWN
+
+    with pytest.raises(TypeError):
+        KafkaError(code=KafkaError._ALL_BROKERS_DOWN)
+
+
 def test_kafkaError_unknown_error():
     with pytest.raises(KafkaException, match="Err-12345?") as e:
         raise KafkaError(12345)