Fix ubuntu issue