Fixed but w/ k_nn and generate and load test works