Fixed bug (shame on me)