Task1 fix tests

Fix tests