diff --git a/company-meghanada.el b/company-meghanada.el
index a04aaff..0ad48ca 100644
--- a/company-meghanada.el
+++ b/company-meghanada.el
@@ -199,9 +199,9 @@
         (return-t (get-text-property 0 'return-type arg)))
     (unless (meghanada--import-exists-p meta)
       (if company-meghanada-auto-import
-          (meghanada--add-import meta)
+          (meghanada--add-import meta (current-buffer))
         (when (y-or-n-p (format "Add import %s ? " (meghanada--import-name meta)))
-          (meghanada--add-import meta))))
+          (meghanada--add-import meta (current-buffer)))))
 
     (save-excursion
       (forward-char -1)