diff --git a/src/analysis.ts b/src/analysis.ts
index 795dc6a..735f213 100644
--- a/src/analysis.ts
+++ b/src/analysis.ts
@@ -47,9 +47,9 @@ type Counts = {
  * @returns Collection with rarity data
  */
 export const analyse = (
-  t: Array<Record<string, string | number | Attribute[]>>
+  c: Array<Record<string, string | number | Attribute[]>> | { tokens: Array<Record<string, string | number | Attribute[]>> }
 ): Collection => {
-  const tokens: Token[] = preprocess(t)
+  const tokens: Token[] = preprocess('tokens' in c ? c.tokens : c)
   const n = tokens.length
   let traitTypes = getTraitTypeSet(tokens)
 
diff --git a/src/scripts/analyse.ts b/src/scripts/analyse.ts
index 337b69d..84bd79b 100644
--- a/src/scripts/analyse.ts
+++ b/src/scripts/analyse.ts
@@ -25,7 +25,7 @@ const argv = yargs(process.argv.slice(2)).options({
 if ('json' in argv) {
   console.log('Analysing data...')
   const rawData = readFileSync(argv.json, { encoding: 'utf-8' })
-  const tokens: Token[] = JSON.parse(rawData)
+  const tokens: Token[] | { tokens: Token[] } = JSON.parse(rawData)
   const collection = analyse(tokens)
 
   if (argv.saveJson) {