Add additional assertion for error message