fix weight layout