changed to py.test syntax