diff --git a/library/localization.py b/library/localization.py
index ff3f96f..9168432 100644
--- a/library/localization.py
+++ b/library/localization.py
@@ -94,7 +94,7 @@ def visualize(im, bbox_cords):
     mask[ymin:ymax,xmin:xmax] = 1
     new_im = new_im*mask
     new_im = new_im.astype(np.uint8)
-    new_im = cv2.cvtColor(new_im, cv2.cv.CV_BGR2RGB)
+    new_im = cv2.cvtColor(new_im, cv2.COLOR_BGR2RGB)
     plt.imshow(new_im)
     
 def calculate_area(c):
@@ -153,4 +153,4 @@ def main():
     bbox_coords = bbox(im,model,layer,n_neurons,kmax)
     #Visualize the Box on the original image
     visualize(im, bbox_cords)
-    return bbox_cords
\ No newline at end of file
+    return bbox_cords