Remove duplicaiton of expectation in test