Fix error look and feel