Fix bug with train units count separat