fix type hint for extend