Test refactoring a bit

Signed-off-by: Geoff Lane <geoff@zorched.net>