Update testing code, fix #33, again.

PR #37 残留了一个它本来想修复的 bug，杯具。