Fix tests to run with changes