A Bug fix in the separation of train\test data