Fix above test