TFJ-775-Small change to fix the bug

TFJ-775