fix type annotations