WIP: Remove WIP from tests