Fixes classes removed