Small code style fixes. Thanks @joelwallis.