Bei der Bundestagswahl 2017 verpasste Wallstein auf Platz 7 der Landesliste der SPD Brandenburg den Einzug in den Bundestag.[6] 2019 übernahm Wallstein den Wahlkampf der SPD Brandenburg zur Europawahl 2019, nachdem Spitzenkandidat Simon Vaut zurückgetreten war.[3] Bei der Bundestagswahl 2021 trat Wallstein im Bundestagswahlkreis Cottbus – Spree-Neiße an, gewann diesen mit 27,6 Prozent der Erststimmen und vertritt damit den Wahlkreis 64 (Cottbus – Spree Neiße) im 20. Deutschen Bundestag.[7]
Trivia
Jedes Jahr läuft Wallstein Hunderte Kilometer zu Fuß mit einem Bollerwagen durch ihren Wahlkreis in der Lausitz.[8]