Fixing missing assert in tests