test: fixed tests after renaming fit to run test: fixed test after refactoring test: added more tests