Shilpa Shetty plans to make a comeback to the silver screen only when her son Viaan starts to understand about her work.