fix to generators tests