Fixed the size bug