add test for new for logic.