tweak to get tests to pass