Change type hint to ObjectManager