forward propagation bug fixed