Dropout bugfix: scale the weights instead of input on the test pass