Wrote missing tests