Fix a little bit more style