diff --git a/src/spyglass/common/common_behav.py b/src/spyglass/common/common_behav.py index d362f3a6c..68f17c156 100644 --- a/src/spyglass/common/common_behav.py +++ b/src/spyglass/common/common_behav.py @@ -226,7 +226,11 @@ def fetch_nwb(self, *attrs, **kwargs) -> list: """ Returns a condatenated list of nwb objects from RawPosition.PosObject """ - return self.PosObject().fetch_nwb(*attrs, **kwargs) + return ( + self.PosObject() + .restrict(self.restriction) # Avoids fetch_nwb on whole table + .fetch_nwb(*attrs, **kwargs) + ) def fetch1_dataframe(self): """Returns a dataframe with all RawPosition.PosObject items.