fixes now failing unit test :D