Another small fix

Another small fix