Yes, as Rashmi says, it’s still mixed up, but in a different way. The first time you had the correct A, but the wrong W. Second time around, you’ve got the correct W, but now the A value is wrong.
Here’s a thread which shows you how to do the “dimensional analysis”, so that you know what should be happening at each layer. And this is just the forward case, which is supposed to be the easy one. 
Things get trickier when we’re going backwards, so it helps to have a “roadmap”. 