Fix typehint in example code