Add missing tests

DESC: Bugfix