Added missing tests

It should be a good practice to add tests with a bug fixes!