Simplify test for demo