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