Fixed a small logical idea