Fixed testing case
Closes #4. Reverted tests