Fix deprecated assertTokens function (use assertTokenTypes though)