fix bug in test, according to Dominic's note