Bug Fix

Signed-off-by: Nazmul Alam <Nazmul.Alam@students.ittralee.ie>