Improved expectation test case