Revert "Fix an attribute error"

This reverts commit 6f127a112a386a6fc35848bad6df681dcd20416f.