Fixed to comply with pytest good practices