Fix one failing test case
Merge pull request #58 from Randy-Ma/fixing_one_test_case

Fix one failing test case