Yes, flax will work great in this recipe and have the same effect as the ground chia seeds. Simply replace the chia seeds with 2 tablespoons of flax meal. Follow the same procedure of allowing the flax to sit in the water for five minutes before mixing it into the dry ingredients.