Add fuller test case