need to think about integration testing to get tests to work properly here