fix iterator type