Quick code style fix