fix style a bit