Make WFI test more strict