small fix from original repo