Added test wrapper to method