Add missing typehint