Fix tests(?)