Fix formatting a little bit