Fixed issue with forward inference and derivative message.