fixed a bug in prepared dict. Will try on full data