Fix problem with script reference