can we move this test stuff out into its own file?