fix small fix