broke infinite loop in back prop