Fixed bug in tests

This is the simple fix for the development branch. Closes #19