Fix code style to match that of surrounding code