fix loop, add test