Niemi pelasi Suomen paidassa vuoden 2009 nuorten MM-kilpailuissa, joissa hän toimi Leijonien varakapteenina. Turnauksessa hän teki tehot 2+1=3. Niemi pelasi myös vuoden 2010 nuorten MM-kilpailuissa toimien Suomen kapteenina ja teki kuudessa ottelussa tehot 3+2=5 ollen Suomen tehokkain puolustaja Sami Vatasen kanssa.