small fix work