fixed tests for small and big datasets