clean up type comparison in unit test