Fix error
small layout