Fixing minor issue regarding code styling