diff --git a/get_train_annot_YOLO.py b/get_train_annot_YOLO.py
index f3efb59d22ffc7e452d331a730b11f48bc5fa83c..e3d3a2df6388b66be2582b11da884db477ed69fe 100644
--- a/get_train_annot_YOLO.py
+++ b/get_train_annot_YOLO.py
@@ -167,12 +167,12 @@ def process(x):
             for l in range(len(fin)):
 
                 try :
-                    plt.gca().add_patch(Rectangle(((fin.x.iloc[l]*len(time))-(0.5*fin.w.iloc[l]*len(time)), ((fin.y.iloc[l]*len(freq))+0.5*fin.height.iloc[l]*len(freq))), fin.width.iloc[l]*len(time), fin.height.iloc[l]*len(freq), linewidth=3, edgecolor=color[int(fin.id.iloc[l])], facecolor='none'))
+                    plt.gca().add_patch(Rectangle(((fin.x.iloc[l]*len(time))-(0.5*fin.width.iloc[l]*len(time)), ((fin.y.iloc[l]*len(freq))+0.5*fin.height.iloc[l]*len(freq))), fin.width.iloc[l]*len(time), fin.height.iloc[l]*len(freq), linewidth=3, edgecolor=color[int(fin.id.iloc[l])], facecolor='none'))
 
                 except IndexError:
                     ipdb.set_trace()
                 try:
-                    plt.savefig(os.path.join(directrory, str('images_annotes_'+str(today.day)+'_'+str(today.month)),str(name+'.jpg')))
+                    plt.savefig(os.path.join(directory, str('images_annotes_'+str(today.day)+'_'+str(today.month)),str(name+'.jpg')))
 
                 except Exception:
                     os.mkdir(os.path.join(directory,str('images_annotes_'+str(today.day)+'_'+str(today.month))))