fix a small logic-problem