Fix wheel size bug