diff --git a/addon.py b/addon.py
index 7df153c..d773514 100644
--- a/addon.py
+++ b/addon.py
@@ -1,5 +1,6 @@
 import sys
 
+import xbmcvfs
 import urllib.request, urllib.parse, urllib.error
 import urllib.parse
 import xbmcgui
@@ -13,7 +14,7 @@
 base_url = sys.argv[0]
 
 __addon__       = xbmcaddon.Addon(id='plugin.program.shortlist')
-__addondir__    = xbmc.translatePath( __addon__.getAddonInfo('profile') )
+__addondir__    = xbmcvfs.translatePath( __addon__.getAddonInfo('profile') )
 __language__     = __addon__.getLocalizedString
 
 addon_handle = int(sys.argv[1])
diff --git a/database.py b/database.py
index e7f2d82..1aca538 100644
--- a/database.py
+++ b/database.py
@@ -1,12 +1,12 @@
 import sys
 import os
 import pickle
-import xbmc
+import xbmcvfs
 import xbmcaddon
 
 # Database location from plugin info
 __addon__       = xbmcaddon.Addon(id='plugin.program.shortlist')
-__addondir__    = xbmc.translatePath( __addon__.getAddonInfo('profile') )
+__addondir__    = xbmcvfs.translatePath( __addon__.getAddonInfo('profile') )
 
 # Settings to see which database
 __database__	= __addon__.getSetting('database').lower()