fix DB mock for testing