Fix to source code styling