Added a test to cover the figure element scenario