figure problem fixed