Fixing bug in finding the selected album