Fix script to run on Ubuntu