fix2

Signed-off-by: Yashar Ahmedizadeh <yashade2001@gmail.com>