Assertion moved to test class