Fix silly typo

Signed-off-by: Aniket Pant <me@aniketpant.com>
Update methods in compliance to tests

Signed-off-by: Aniket Pant <me@aniketpant.com>
Add tests

Signed-off-by: Aniket Pant <me@aniketpant.com>