Switch to assertj asserts