Fix controllers tests