Code style fix

Just making it consistent with other code :)