Change tests to assert elements