Missing class fix