Lazy fix to test