Fix tests for various elements