Fixes failed test case