Adds missing tests