Bug fix added missing dependency