Fixed roots test

Temporary fix. Must come back to this!