Trying to fix this change to the private API