fix tests srsly