fix type annotation