A little bug fix

A little bug fix