Citation: 1B_555/2021 E. C

Gegen den Entscheid des Obergerichts vom 7. September 2021 hat A.________ am 6. Oktober 2021 Beschwerde in Strafsachen an das Bundesgericht erhoben. Er beantragt, der angefochtene Entscheid sei aufzuheben und die Angelegenheit zur Neubeurteilung an das Obergericht zurückzuweisen. Eventualiter seien der angefochtene Entscheid und die Dispositiv-Ziffern 5-8 des Urteils des Bezirksgerichts vom 22. Juni bzw. 26. Juli 2021 aufzuheben und ihm eine Entschädigung gemäss kantonalem Anwaltstarif, mindestens in Höhe von Fr. 4'000.-- (einschliesslich Barauslagen und Mehrwertsteuer), zuzusprechen. Die Vorinstanz und die Generalstaatsanwaltschaft haben auf Vernehmlassung verzichtet und beantragen die Abweisung der Beschwerde.