Improved KDB unit test a bit