Fix: Missing class