fix: tests according to previous commit