Change to assertThat