Add missing dependency
Split out separate test widget