Tiny change to include whole device