Fixed a failing test related to #54