Fix pytest-one

This function wasn't working due to a missing argument.