Fix up unicorn script