Renaming decorator and removing double import in test file.