Added missing argument to tests. It's Friday afternoon