Test ModelGraph remove.
Fix model graph removal.