Fix compile  error