fix type; add test