Remove WIP from tests