Fixed the size issue