Fix docs script