Oops...remove '.py' from YAML