Fix how tcn test considers total num classes automatically

Fix how tcn test considers total num classes automatically #99