tests: Make compatible with both KVM and UML