MINOR CHANGE: Missing class