Für Finnland nahm Niemi an den U20-Junioren-Weltmeisterschaften 2009 und 2010 teil. Beim Turnier 2009 war er Assistenzkapitän, ein Jahr später Mannschaftskapitän.