Johan von Wallenstein, greve