Image bug fix