Small change on test case