Made public test set load correctly