Small fix on db