Add basic test for new helper method