Fix this recently moved code to use the correct type. CI is now a