Update tests to work with new compiler