Fix timeout test