The line ```py scale_wise_hists = [hists[3 * j + i] for j in range(len(hists) // num_scale_factors)] ``` is not in a general form. (only for 3 scale factors)