Attempting to unit test