Sweden 4, Czechia 2 — Herman Traff (New Jersey Devils) scored two power-play goals as Sweden (4-0-0-0) finished first in Group B at TD Place.

Felix Unger Sorum (Carolina Hurricanes) and Anton Wahlberg (Buffalo Sabres) also scored for Sweden, which will play Latvia in the quarterfinals on Thursday. Melker Thelin (Utah Hockey Club) made 28 saves.

“We played like we want to play,” Sweden defenseman Axel Sandin-Pellikka (Detroit Red Wings) said. “We held our structure, played quick from the back end, and got some shots on net. It was a good second period.”

Petr Sikora (Washington Capitals) and Eduard Sale (Seattle Kraken) scored for Czechia (3-0-0-1), which finished second in the group and will play the third-place finisher from Group A. Michal Hrabal (Utah Hockey Club) made 37 saves.

“It’s tough after this game to say something, because we had so many penalties,” Sale said. “They are a great team, and they have a great power play. So we had, like, seven PK [situations], so it was too much in this game.”

Traff, who turned 19 on Tuesday, made it 1-0 on the man-advantage at 1:52 of the second period. Theo Lindstein (St. Louis Blues) had his shot blocked, but the puck bounced to Traff, who scored from the high slot.

Sweden went ahead 2-0 at 7:18 when Traff scored a power-play goal off the rebound of a shot by Otto Stenberg (St. Louis Blues).

“The birthday boy, Herman … incredible game for him,” Sandin-Pellikka said. “So extra happy for him.”

Sikora scored a power-play goal on a one-timer from the right face-off circle to make it 2-1 at 11:57.

Sweden pushed the lead to 4-1 with goals 30 seconds apart by Unger Sorum and Wahlberg.

Unger Sorum’s shot from the right circle went off the stick of Czechia defenseman Jakub Dvorak (Los Angeles Kings) and over Hrabal’s glove at 18:17, and then Wahlberg scored from in front at 18:47.

Sale was slashed by Sweden defenseman Axel Sandin-Pellikka (Detroit Red Wings) on a breakaway and was awarded a penalty shot 34 seconds into the third period. Sale, Czechia’s captain, skated in on Thelin, deked, and lifted a backhand over the goalie’s left pad to make it 4-2.

Times for the quarterfinal game are still to be determined.