fix: add type argument