Minor DOM fix