Changed assertions to match test description