Quick code format fix