Fixed bug with add button not working for album