Fix spec, wasn't working after first run