Für die Herren-Nationalmannschaft gab Vainshtain sein Debüt bei der Weltmeisterschaft 2013, als ihm mit seiner Mannschaft in der Division II der Aufstieg von der Gruppe B in die Gruppe A gelang. Auch bei der Weltmeisterschaft 2015 spielte er in der Division II. Zudem vertrat er seine Farben bei der Olympiaqualifikation für die Winterspiele 2018 in Pyeongchang.