Fixed tests to use assert