Change return typehint