small fix but not complete