Update element_array_ephys/export/nwb/nwb.py

Co-authored-by: Kabilar Gunalan <kabilar@datajoint.com>

Author: bendichter

element_array_ephys/export/nwb/nwb.py

@@ -215,10 +215,10 @@ def create_units_table(
     ):
 
         probe_id, shank_num = (
-            probe.ProbeType.Electrode
-            * ephys.ProbeInsertion
+            ephys.ProbeInsertion
             * ephys.CuratedClustering.Unit
-            & {"unit": unit["unit"], "insertion_number": unit["insertion_number"]}
+            * probe.ProbeType.Electrode
+            & unit
         ).fetch1("probe", "shank")
 
         waveform_mean = (