fix a bug ...