this is just fixing existing tests. it's missing a test for the added behaviour.