Fixed album edit function