fix minor bug, again.