fix TypeError: 'int' object is not iterable

fix "TypeError: 'int' object is not iterable" error