small logical fix