Small adjust in new version