Forgot to add some simple tests