Fixed error on a test case