Only test what we care about for now