Fix test

This still needs to be… tested