fixing tests, but weird error here