Updated test case to have proper asserts.