Fix mestake at tests