Fix for db class and the optional db parameter