fix regression in unit-checking feature