diff --git a/partitioncloud/modules/albums.py b/partitioncloud/modules/albums.py
index aa5500f..6d86464 100644
--- a/partitioncloud/modules/albums.py
+++ b/partitioncloud/modules/albums.py
@@ -4,6 +4,7 @@ Albums module
 """
 import os
 import shutil
+from uuid import uuid4
 
 from flask import (Blueprint, abort, flash, redirect, render_template, request,
                    send_file, session, current_app)