fix failing tests [Closes #5]