fix code of equation