Add import for ProblemDetector and change actor lookup to suit.