Adding tests for the helper