Fixing broken typehint