FIxed to test correctly