Fix timeout causing a failed test