Fixed class cast and imports.