chore(tests): Reenable all the tests and fix a bug in tests