Update tests to not error out