Removed unneeded import of wraps