fix broken feature tests