oops, fix type annotations