one more logic fix