fix display issue, no images yet though :(