Fixed cs error

Fixes #4