update to work on ubuntu