Fix test
Simplify test