fix output layer bug with output activation