fix: Tests now works fine