Fix of shape mismatch

4 jobs for master in 11 minutes and 57 seconds (queued for 1 second)