Fix regression caused by previous tests