Fix sizing bug