Fix bug that fail test (Related by #3).