Fix error layout

Closes #4