Another small fix