Fixes a missing class issue