fixed the failing test for partition in inmemory mode