Fix unicorn script issue