fix: regexp in train command