Update method calls for new return type