diff --git a/litex/gen/genlib/record.py b/litex/gen/genlib/record.py
index e98922ffd..2692ab4b6 100644
--- a/litex/gen/genlib/record.py
+++ b/litex/gen/genlib/record.py
@@ -136,10 +136,14 @@ class Record:
             _keep = set([f[0] for f in self.layout])
         elif isinstance(keep, list):
             _keep = set(keep)
+        else:
+            _keep = keep
         if omit is None:
             _omit = set()
         elif isinstance(omit, list):
             _omit = set(omit)
+        else:
+            _omit = omit
 
         _keep = _keep - _omit