Fix skipped unit test