* fix small logic error