Trello #459

Fix for unit test