Setup was replaced with class as an option in the initializer - support both