Fix broken code caused by attempted removal of type annotations.