Fix for DB update