Fix a small problem about complexity counting