Fix a broken test, Added a missing returning variable