fix tests

should this be a diff file?