Please also add typing for the utility methods in tests.