Fix typhints