Some small fix

Some small fix