Fix black image issue