Bug fix

Signed-off-by: tiendam <tien.dam@socketlabs.com>