Fixed issue with pytest