Missing return added