Adding return test.