Fixed: rewrite test case