Fix staff tests