diff --git a/Sinvo.EplanHpD.Plugin.WPFUI/View/LectotypeWindow.xaml b/Sinvo.EplanHpD.Plugin.WPFUI/View/LectotypeWindow.xaml
index 077bc86..af3922a 100644
--- a/Sinvo.EplanHpD.Plugin.WPFUI/View/LectotypeWindow.xaml
+++ b/Sinvo.EplanHpD.Plugin.WPFUI/View/LectotypeWindow.xaml
@@ -276,7 +276,8 @@
                                                 <Button
                                                     Margin="10,0,0,0"
                                                     Click="MotorListViewToMotorSourceMenuItem_Click"
-                                                    Content="查看3D中电机" />
+                                                    Content="查看3D中电机"
+                                                    Tag="{Binding OccPartId}" />
                                             </StackPanel>
                                         </Grid>
                                     </Expander.Header>
diff --git a/Sinvo.EplanHpD.Plugin.WPFUI/View/LectotypeWindow.xaml.cs b/Sinvo.EplanHpD.Plugin.WPFUI/View/LectotypeWindow.xaml.cs
index 8419858..74532fe 100644
--- a/Sinvo.EplanHpD.Plugin.WPFUI/View/LectotypeWindow.xaml.cs
+++ b/Sinvo.EplanHpD.Plugin.WPFUI/View/LectotypeWindow.xaml.cs
@@ -349,22 +349,10 @@ namespace Sinvo.EplanHpD.Plugin.WPFUI
 
         private void MotorListViewToMotorSourceMenuItem_Click(object sender, RoutedEventArgs e)
         {
-            var selectItem = MotorListView.SelectedItem;
-            if (selectItem is MotorModel motor)
+            if(sender is Button btn)
             {
-                var motorModelPartId = motor?.OccPartId;
-
-                //var cableName = model.IsComplexLine ? model.CableName.Split('/')[1] : model.CableName;
-                //if (string.IsNullOrEmpty(motorModelPartId))
-                //{
-                //    FlexMessageBox.ShowText(FlexMessageBox.Type.INFO, "未获取到电机名称");
-                //}
-                //else
-                {
-
-                    ViewModel.ToMotorSource(motorModelPartId, "");
-                }
-                //}
+                var motorModelPartId = btn.Tag as string;
+                ViewModel.ToMotorSource(motorModelPartId, "");
             }
         }
 
@@ -372,7 +360,6 @@ namespace Sinvo.EplanHpD.Plugin.WPFUI
         private void StartLayoutBtn_Click(object sender, RoutedEventArgs e)
         {
             var motorIds = ViewModel.Motors.Where(motor => !motor.IsError).Select(motor => motor.OccPartId).ToList();
-
             var window = new LayoutHelperWindow(motorIds);
             //window.MotorIds = motorIds;
             ElementHost.EnableModelessKeyboardInterop(window);