added assertion to test