Update to remove deprecated function