Fix new class dep