From a43dbb1a4f57e766f41e4d0db837dbdf82e8c21d Mon Sep 17 00:00:00 2001
From: Niels <niels.muendler@inf.ethz.ch>
Date: Tue, 10 Sep 2024 16:50:48 +0200
Subject: [PATCH] Fix checks

Signed-off-by: Niels <niels.muendler@inf.ethz.ch>
---
 lib/Command/Classify.php | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/lib/Command/Classify.php b/lib/Command/Classify.php
index ae4c8fa6..75832e26 100644
--- a/lib/Command/Classify.php
+++ b/lib/Command/Classify.php
@@ -109,16 +109,16 @@ protected function execute(InputInterface $input, OutputInterface $output): int
 					MusicnnClassifier::MODEL_NAME => [],
 				];
 				foreach ($this->storageService->getFilesInMount($mount['storage_id'], $mount['override_root'], $models, $lastFileId) as $file) {
-					$i++;
-					$lastFileId = $file['fileid'];
 					// if retry flag is set, skip tagged files
 					if ($input->getOption('retry')) {
 						$fileTags = $this->tagManager->getTagsForFiles([$lastFileId]);
 						// check if processed tag is already in the tags
-						if (in_array($processedTag, $fileTags)) {
+						if (in_array($processedTag, $fileTags[$lastFileId])) {
 							continue;	
 						}
 					}
+					$i++;
+					$lastFileId = $file['fileid'];
 					$queueFile = new QueueFile();
 					$queueFile->setStorageId($mount['storage_id']);
 					$queueFile->setRootId($mount['root_id']);