fixing tests yo