Fix layout measure issue