Small fix to assert