Fix assertion

This closes #13 noting the problem.