fix dis bug