Add unit-tests mistakenly ommitted