Fix unstable test