minor bug fix : class image