Fixed little code styling issue