Fixes #2 layout bug