fix tests && add test