fix: PSI train only in base 1

fixed: Bug report #124 - PSI train only in base 1