Fixes DB test