Remove deprecation warning
Switch to test/unit for compatibility