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);
});
}