Fixes my tests