Fix a bug

Signed-off-by: MKRussel <russel@smitsol.com>