Fix test to not rely on missing def