Remove unnecessary checks in tests