Fix data forwarding bug
Fix mem stage bug