fixing bug - must return the component