Fixed typo in arguments
Added assertion test