Making test more stable