Fix assertion