Fixed unhandled iteration on NoneType