Add some missting tests