Fixing unit test (botched merge)