update to compile with latest UniMath