Cleaned up pytest annotations.