Small fix/formatting