You don't have to change this, but I'm almost always against unit tests against an internal API.  Just something to keep in mind for later.