fix broken test case assertion