Fix. the forget append last layer