Updated wrapper