diff --git a/spec/models/budget_spec.rb b/spec/models/budget_spec.rb
index fa0c6ed57..cbc5fd5de 100644
--- a/spec/models/budget_spec.rb
+++ b/spec/models/budget_spec.rb
@@ -123,13 +123,13 @@ describe Budget do
     it "returns nil if there is only one budget and it is still in drafting phase" do
       budget = create(:budget, phase: "drafting")
 
-      expect(Budget.current).to eq(nil)
+      expect(described_class.current).to eq(nil)
     end
 
     it "returns the budget if there is only one and not in drafting phase" do
       budget = create(:budget, phase: "accepting")
 
-      expect(Budget.current).to eq(budget)
+      expect(described_class.current).to eq(budget)
     end
 
     it "returns the last budget created that is not in drafting phase" do
@@ -138,7 +138,7 @@ describe Budget do
       current_budget  = create(:budget, phase: "accepting", created_at: 1.month.ago)
       next_budget     = create(:budget, phase: "drafting",  created_at: 1.week.ago)
 
-      expect(Budget.current).to eq(current_budget)
+      expect(described_class.current).to eq(current_budget)
     end
 
   end
@@ -146,10 +146,9 @@ describe Budget do
   describe "#open" do
 
     it "returns all budgets that are not in the finished phase" do
-      phases = Budget::PHASES - ["finished"]
-      phases.each do |phase|
+      (Budget::Phase::PHASE_KINDS - ["finished"]).each do |phase|
         budget = create(:budget, phase: phase)
-        expect(Budget.open).to include(budget)
+        expect(described_class.open).to include(budget)
       end
     end