Missed two tests!

(Mentioned in #58, thanks!)