Fixing embarassing code typo

Co-authored-by: Wojciech Ziolek <wojciech.ziolek@cern.ch>