Small aesthetic fix