fixed test (missed out one line in commit e80d56d)