Minor: fixed graph closing on MT env
Removed unused class