stage 3 - fix a div error