Fix DB tests