Fix unit testing with openfl 2.0