Tests were failing previously but now it looks fine without this guard.