Use correct asserts
Maybe?