refactor: fix tests to check visibility correctly