Fix CS
Add test