Make assertion work as intended