#31 Add component in setup
Added missing mock annotation #31