Fixing simple bug ;)