Fix buge in previous commit. cc #134.