Added some missing assertions :)