fix an issue now the mod work fully!!