diff --git a/lib/poseidon/topic_metadata.rb b/lib/poseidon/topic_metadata.rb
index 3265c4b..bdd96ec 100644
--- a/lib/poseidon/topic_metadata.rb
+++ b/lib/poseidon/topic_metadata.rb
@@ -56,7 +56,7 @@ def partition_count
 
     def available_partitions
       @available_partitions ||= struct.partitions.select do |partition|
-        partition.error == 0 && partition.leader != -1
+        (partition.error == Errors::NO_ERROR_CODE || Errors::ERROR_CODES[partition.error] == Errors::ReplicaNotAvailable) && partition.leader != -1
       end
     end