Fix type check error
Fix type check error
Fix invalid type annotation