Fix test, add template