Add fig to test