Small style fix