Minor coding fix.
Modified the db wrapper for compliance with the modifications.