Small fixes
Small fix to figure element