Somehow not working. Maybe missing some test cases