Added test case fix

fix #757