Correct docblock type hint for set_db($db)