Fix bad coding style, resolves #2