should probably update the test too