Touched up assertion and everything is looking good