Fix test
Working test