Parece que você está se aproximando desse problema corretamente. Eu faria do mesmo jeito. Você pode compartilhar alguns exemplos da mensagem de erro? Qual parte do script está jogando o erro?
Eu gostaria de comentar sua pergunta acima, mas não tenho a reputação necessária.