Fix bugs train on spam data