Fix error in test_table_keys_per_column with some modifications to DemoDB