Fix second test

Added assert to second test