small code patch for the bug fix (VER: 0.3.1)