[#55] Update tests
[#55] Update tests