fix: tests 2