Fix instance class check of TestDriver