Fix style a bit more