Slight style fix