I was planning on adding more tests around this once the UI is all set.