small code fix
fixed small problem