fix weight saving