Training and testing are now separated :-)