diff --git a/QuickSVGExport.FCMacro b/QuickSVGExport.FCMacro
index d91a0b6..d7095a7 100644
--- a/QuickSVGExport.FCMacro
+++ b/QuickSVGExport.FCMacro
@@ -1,5 +1,6 @@
 import importSVG
 import Draft
+import FreeCAD
 
 outputRacine = App.activeDocument().getFileName().rpartition('.')[0]
 
@@ -29,16 +30,15 @@ for selectedObject in Gui.Selection.getSelectionEx():
 	try:
 		if hasattr(selectedObject.Object, 'Dir'):
 			sv0 = Draft.make_shape2dview(selectedObject.Object, FreeCAD.Vector(selectedObject.Object.Dir))
-			sv0.ViewObject.LineColor = (1.0, 0.0, 0.0)
 		elif hasattr(selectedObject.Object, 'Objects'):
 			for child in selectedObject.Object.Objects:
 				if hasattr(child, 'Dir'):
 					sv0 = Draft.make_shape2dview(selectedObject.Object, FreeCAD.Vector(child.Dir))
-					sv0.ViewObject.LineColor = (1.0, 0.0, 0.0)
 					break
 			print("FAILURE IF")
 		FreeCAD.ActiveDocument.recompute()
 		sk = Draft.make_sketch(sv0, autoconstraints=True)
+		sk.ViewObject.LineColor = (1.0, 0.0, 0.0)
 		FreeCAD.ActiveDocument.recompute()
 		sketchList.append(sk)
 		if hasattr(sv0, 'Name'):
@@ -46,4 +46,4 @@ for selectedObject in Gui.Selection.getSelectionEx():
 	except:
 		print("FAILURE TRY")
 
-exportSketch(sketchList)
+exportSketch(sketchList)
\ No newline at end of file