Add more test :metal: