fixed the size problem