fixing bad fix on test case