add missing type annotation