Shanchen

Small fix, that corrects some wrong code left by mistake after debug.