diff --git a/examples/cpp/diffusion/src/main.cu b/examples/cpp/diffusion/src/main.cu
index 58d60ebce..089a00212 100644
--- a/examples/cpp/diffusion/src/main.cu
+++ b/examples/cpp/diffusion/src/main.cu
@@ -137,7 +137,6 @@ int main(int argc, const char ** argv) {
 
     /**
      * Create visualisation
-     * @note FLAMEGPU2 doesn't currently have proper support for discrete/2d visualisations
      */
 #ifdef FLAMEGPU_VISUALISATION
     flamegpu::visualiser::ModelVis visualisation = cudaSimulation.getVisualisation();
@@ -149,6 +148,8 @@ int main(int argc, const char ** argv) {
         visualisation.setCameraSpeed(0.001f * SQRT_AGENT_COUNT);
         visualisation.setViewClips(0.01f, 2500);
         visualisation.setClearColor(0.0f, 0.0f, 0.0f);
+        visualisation.setOrthographic(true);
+        visualisation.setOrthographicZoomModifier(0.284f);
         auto agt = visualisation.addAgent("cell");
         // Position vars are named x, y, z; so they are used by default
         agt.setModel(flamegpu::visualiser::Stock::Models::CUBE);  // 5 unwanted faces!