Fix square movement after integration