bug fix for final image