Small bug fix.

More sensible layout.