Write a few more tests for coverage (#501)

2 files changed