Updated test slightly to reflect button