Fixing borken test