Fix for labels-outputs NN swap