Fix for pycept naming changes