Chiefs focused on set piece