fix style a little bit