Fix bug
Update test