use last layer, not first