public static void WaitReload(this IEntityButtonContainer container, Action action)
    {
        var oldCount = container.RefreshCount() !.Value;

        action();
        container.Element.GetDriver().Wait(() =>
        {
            var newCount = container.RefreshCount();
            return(newCount != null && newCount != oldCount);
        });
    }