fix typo in train.m