output layer fixes