Adds a more flexible test