Fix

fix logic error