domain: fix tests

cc/ @jrwren looks like this is what you were trying to do here –
let me know if that's not right. Sorry for the delay!