fix weight init