Minor changes to see what happen with the tests