diff --git a/supabase/lib/realtime_client.py b/supabase/lib/realtime_client.py
index 83fe0173..22792f11 100644
--- a/supabase/lib/realtime_client.py
+++ b/supabase/lib/realtime_client.py
@@ -15,10 +15,10 @@ def __init__(self, socket: Socket, schema: str, table_name: str):
 
     def get_payload_records(self, payload: Any):
         records: dict = {"new": {}, "old": {}}
-        if payload.type == "INSERT" or payload.type == "UPDATE":
+        if payload.type in ["INSERT", "UPDATE"]:
             records["new"] = payload.record
             convert_change_data(payload.columns, payload.record)
-        if payload.type == "UPDATE" or payload.type == "DELETE":
+        if payload.type in ["UPDATE", "DELETE"]:
             records["old"] = payload.record
             convert_change_data(payload.columns, payload.old_record)
         return records