Minor fix to equations