Update for new pykit builder