Fix the ubuntu dockerfile