Fixed assertThat import