Fix style a bit