diff --git a/generate.py b/generate.py
index 9270063..0b9561b 100755
--- a/generate.py
+++ b/generate.py
@@ -11,7 +11,7 @@ config = json.load(open(os.path.expanduser('~/.ox/client.json')))
 api.signin(username=config['username'], password=config['password'])
 
 def get_files(item):
-    files = api.findFiles({
+    files = api.findMedia({
         "keys": [
             "selected",
             #"path",