Updated DB tests

Removed unused code