Fixed a little issue
Updated
Update test