diff --git a/test/docs.jl b/test/docs.jl
index b7efea07..7338be12 100644
--- a/test/docs.jl
+++ b/test/docs.jl
@@ -31,7 +31,7 @@
         #print(pm.model)
 
         @test JuMP.num_nonlinear_constraints(pm.model) == 0
-        @test JuMP.num_constraints(pm.model, JuMP.NonlinearExpr{JuMP.VariableRef}, JuMP.MOI.EqualTo{Float64}) == 12
+        @test JuMP.num_constraints(pm.model, JuMP.NonlinearExpr, JuMP.MOI.EqualTo{Float64}) == 12
         @test JuMP.num_variables(pm.model) == 28
 
         result = optimize_model!(pm, optimizer=JuMP.optimizer_with_attributes(Ipopt.Optimizer, "print_level"=>0))
diff --git a/test/model.jl b/test/model.jl
index a67b3497..77ad617b 100644
--- a/test/model.jl
+++ b/test/model.jl
@@ -16,7 +16,7 @@
         pm = instantiate_model("../test/data/matpower/case5.m", ACPPowerModel, PowerModels.build_opf, jump_model=m)
 
         @test JuMP.num_nonlinear_constraints(pm.model) == 0
-        @test JuMP.num_constraints(pm.model, JuMP.NonlinearExpr{JuMP.VariableRef}, JuMP.MOI.EqualTo{Float64}) == 28
+        @test JuMP.num_constraints(pm.model, JuMP.NonlinearExpr, JuMP.MOI.EqualTo{Float64}) == 28
         @test JuMP.num_variables(pm.model) == 49
 
         @test pm.model[:my_var] == x
diff --git a/test/opf-obj.jl b/test/opf-obj.jl
index 48b4045c..0f4fff3d 100644
--- a/test/opf-obj.jl
+++ b/test/opf-obj.jl
@@ -69,7 +69,7 @@ end
 
     @testset "jump model objective type" begin
         pm = instantiate_model(data, ACPPowerModel, build_opf)
-        @test JuMP.objective_function(pm.model) isa JuMP.NonlinearExpr{JuMP.VariableRef}
+        @test JuMP.objective_function(pm.model) isa JuMP.NonlinearExpr
     end
 
     @testset "opf objective" begin