Album bug fix