1 files changed, 4 insertions, 1 deletions
diff --git a/src/gpu/effect.h b/src/gpu/effect.h
index 6fdb0f4..8f35f3c 100644
--- a/src/gpu/effect.h
+++ b/src/gpu/effect.h
@@ -1,5 +1,7 @@
 #pragma once
 #include "gpu/gpu.h"
+#include "gpu/effects/post_process_helper.h"
+#include "gpu/uniform_helper.h"
 #include <algorithm>
 #include <map>
 #include <memory>
@@ -12,6 +14,7 @@ class PostProcessEffect;
 class Effect {
  public:
   Effect(const GpuContext& ctx) : ctx_(ctx) {
+    uniforms_.init(ctx.device);
   }
   virtual ~Effect() = default;
   virtual void init(MainSequence* demo) {
@@ -43,7 +46,7 @@ class Effect {
 
  protected:
   const GpuContext& ctx_;
-  GpuBuffer uniforms_;
+  UniformBuffer<CommonPostProcessUniforms> uniforms_;
   int width_ = 1280;
   int height_ = 720;
 };