Added `ms` unit name to skip `transition` optimization

Fixed tests