Add utility method for testing