Fixed wrong datasets splitting.