Fix build error because of missing #import