Fixed little mistake and add new test file