1 files changed, 3 insertions, 2 deletions
diff --git a/encrypt.py b/encrypt.py
index 80cedf3..e813d36 100644
--- a/encrypt.py
+++ b/encrypt.py
@@ -14,6 +14,7 @@ if __name__ == '__main__':
         print(f'{infile} is not a file or directory, exiting.')
         os.exit(2)
 
-    file_id, token = encrypt_file(args.infile)
-    print(f'/{file_id}/{token}/{os.path.basename(args.infile)}')
+    download_filename = os.path.basename(args.infile)
+    file_id, token = encrypt_file(args.infile, download_filename)
+    print(f'/{file_id}/{token}/{download_filename}')