Fixed asserts in test case to be right