small wrapper fix