add missing device argument