Fix bug "Class 'string' not found"