Minor style fixes

Forgot to add it into #44