Fix build for Ubuntu