Fix test after change in mr.awsome.