Correct reshaping in weight method

4 jobs for glspace_fixes in 14 minutes and 38 seconds (queued for 1 second)